lxuechen
/

phi-2-dpo

Text Generation

Inference Endpoints

Model card Files Files and versions Community

lxuechen commited on Jan 11, 2024

Commit

6e73685

·

verified ·

1 Parent(s): 4077ba9

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -48,8 +48,9 @@ Here's a full-fledged example:
 import torch
 import transformers
 model: transformers.PreTrainedModel = transformers.AutoModelForCausalLM.from_pretrained(
-    "lxuechen/phi-2-dpo",
     low_cpu_mem_usage=True,
     device_map="auto",
     trust_remote_code=True,

 import torch
 import transformers
+model_name_or_path = "lxuechen/phi-2-dpo"
 model: transformers.PreTrainedModel = transformers.AutoModelForCausalLM.from_pretrained(
+    model_name_or_path,
     low_cpu_mem_usage=True,
     device_map="auto",
     trust_remote_code=True,