FrankL
/

storytellerLM-v0.1

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

FrankL commited on May 26, 2024

Commit

540469a

·

verified ·

1 Parent(s): c526ddd

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -18,7 +18,7 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 ### Direct Use
 ```python
-model = AutoModel.from_pretrained('FrankL/storytellerLM-v0.1', trust_remote_code=True, torch_dtype=torch.float16)
 model = model.to(device='cuda')
 tokenizer = AutoTokenizer.from_pretrained('FrankL/storytellerLM-v0.1', trust_remote_code=True)

 ### Direct Use
 ```python
+model = AutoModelForCausalLM.from_pretrained('FrankL/storytellerLM-v0.1', trust_remote_code=True, torch_dtype=torch.float16)
 model = model.to(device='cuda')
 tokenizer = AutoTokenizer.from_pretrained('FrankL/storytellerLM-v0.1', trust_remote_code=True)