hugohrban
/

progen2-large

Text Generation

Model card Files Files and versions

hugohrban commited on Jun 9, 2024

Commit

59fe124

·

verified ·

1 Parent(s): 60c3abb

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -14,7 +14,7 @@ import torch
 import torch.nn.functional as F
 # load model and tokenizer
-model = AutoModelForCausalLM.from_pretrained("hugohrban/progen2-large", trust_remote_code=True)
 tokenizer = Tokenizer.from_pretrained("hugohrban/progen2-large")
 tokenizer.no_padding()

 import torch.nn.functional as F
 # load model and tokenizer
+model = AutoModelForCausalLM.from_pretrained("hugohrban/progen2-large", trust_remote_code=True, torch_dtype="auto")
 tokenizer = Tokenizer.from_pretrained("hugohrban/progen2-large")
 tokenizer.no_padding()