lxuechen
/

phi-2-tool-use

Text Generation

Inference Endpoints

Model card Files Files and versions Community

lxuechen commited on Jan 13, 2024

Commit

640d4e3

·

verified ·

1 Parent(s): dc0ca49

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

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

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