Commit
•
c078716
1
Parent(s):
8b97b54
Update README.md
Browse files
README.md
CHANGED
|
@@ -18,7 +18,7 @@ Note: These are only LoRA adapters. You should also import the base model itself
|
|
| 18 |
|
| 19 |
Example usage:
|
| 20 |
|
| 21 |
-
```
|
| 22 |
model_name = "unsloth/Meta-Llama-3.1-8B"
|
| 23 |
model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype=torch.float16)
|
| 24 |
model.gradient_checkpointing_enable()
|
|
|
|
| 18 |
|
| 19 |
Example usage:
|
| 20 |
|
| 21 |
+
```py
|
| 22 |
model_name = "unsloth/Meta-Llama-3.1-8B"
|
| 23 |
model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype=torch.float16)
|
| 24 |
model.gradient_checkpointing_enable()
|