Update README.md
Browse files
README.md
CHANGED
|
@@ -92,6 +92,7 @@ model = Gemma2ForCausalLM.from_pretrained( # Changed here
|
|
| 92 |
device_map="auto",
|
| 93 |
attn_implementation=attn_implementation
|
| 94 |
)
|
|
|
|
| 95 |
tokenizer = GemmaTokenizerFast.from_pretrained(base_model, padding_side="right",
|
| 96 |
truncation_side="right", trust_remote_code=True)
|
| 97 |
|
|
|
|
| 92 |
device_map="auto",
|
| 93 |
attn_implementation=attn_implementation
|
| 94 |
)
|
| 95 |
+
|
| 96 |
tokenizer = GemmaTokenizerFast.from_pretrained(base_model, padding_side="right",
|
| 97 |
truncation_side="right", trust_remote_code=True)
|
| 98 |
|