Chryslerx10
commited on
Commit
•
ed8c139
1
Parent(s):
7f01b78
Update README.md
Browse files
README.md
CHANGED
@@ -61,7 +61,7 @@ Libraries Used:
|
|
61 |
|
62 |
|
63 |
## How to Use
|
64 |
-
|
65 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
66 |
from peft import PeftModel, PeftConfig
|
67 |
|
@@ -78,3 +78,4 @@ Libraries Used:
|
|
78 |
tokenizer.pad_token = tokenizer.eos_token
|
79 |
|
80 |
peft_loaded_model = PeftModel.from_pretrained(model, peft_model_id, device_map='auto')
|
|
|
|
61 |
|
62 |
|
63 |
## How to Use
|
64 |
+
```python
|
65 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
66 |
from peft import PeftModel, PeftConfig
|
67 |
|
|
|
78 |
tokenizer.pad_token = tokenizer.eos_token
|
79 |
|
80 |
peft_loaded_model = PeftModel.from_pretrained(model, peft_model_id, device_map='auto')
|
81 |
+
```
|