Update README.md
Browse files
README.md
CHANGED
|
@@ -45,6 +45,7 @@ from transformers import AutoModelForCausalLM, BitsAndBytesConfig, set_seed
|
|
| 45 |
set_seed(42)
|
| 46 |
|
| 47 |
# Load model
|
|
|
|
| 48 |
modelpath = "DisgustingOzil/phi-2-riddler"
|
| 49 |
model = AutoModelForCausalLM.from_pretrained(
|
| 50 |
modelpath,
|
|
|
|
| 45 |
set_seed(42)
|
| 46 |
|
| 47 |
# Load model
|
| 48 |
+
```python
|
| 49 |
modelpath = "DisgustingOzil/phi-2-riddler"
|
| 50 |
model = AutoModelForCausalLM.from_pretrained(
|
| 51 |
modelpath,
|