lxuechen
/

phi-2-dpo

Text Generation

Inference Endpoints

Model card Files Files and versions Community

lxuechen commited on Dec 27, 2023

Commit

b7b98a1

·

1 Parent(s): d5ebbc5

Update README.md

Files changed (1) hide show

README.md +3 -0

README.md CHANGED Viewed

@@ -45,6 +45,9 @@ where `instruction` is your query.
 Here's a full-fledged example:
 ```
 model: transformers.PreTrainedModel = transformers.AutoModelForCausalLM.from_pretrained(
     "lxuechen/phi-2-dpo",
     low_cpu_mem_usage=True,

 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,