rishiraj
/

Phi-3-mini-4k-ORPO

Text Generation

Model card Files Files and versions Community

rishiraj commited on May 1

Commit

55d9834

•

1 Parent(s): 1d77ab4

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -45,9 +45,9 @@ This code snippets show how to get quickly started with running the model on a G
 pip install peft transformers bitsandbytes accelerate
 ```
 ```python
-from peft import AutoPeftModelForCausalLM
 from transformers import AutoTokenizer
-model = AutoPeftModelForCausalLM.from_pretrained(
     "rishiraj/Phi-3-mini-4k-ORPO",
     load_in_4bit = True,
 )

 pip install peft transformers bitsandbytes accelerate
 ```
 ```python
+from transformers import AutoModelForCausalLM
 from transformers import AutoTokenizer
+model = AutoModelForCausalLM.from_pretrained(
     "rishiraj/Phi-3-mini-4k-ORPO",
     load_in_4bit = True,
 )