DukeNLP
/

Prob-Gen-70B

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

DukeNLP commited on May 20

Commit

24784b4

•

1 Parent(s): a5c4027

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -33,7 +33,7 @@ The model can be loaded with HuggingFace's Transformers library:
 ``` python
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model_id = "DukeNLP/Prob-Gen-8B"
 model = AutoModelForCausalLM.from_pretrained(model_id,device_map="auto", trust_remote_code=True)

 ``` python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model_id = "DukeNLP/Prob-Gen-70B"
 model = AutoModelForCausalLM.from_pretrained(model_id,device_map="auto", trust_remote_code=True)