ddidacus
/

RS-mol-llama-1b

Text Generation

text-generation-inference

Model card Files Files and versions Community

ddidacus commited on Feb 7

Commit

a56d365

·

verified ·

1 Parent(s): 97f9694

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -21,8 +21,8 @@ import re
 # Setup
 device = "cuda"
-tokenizer = AutoTokenizer.from_pretrained("ddidacus/RS-molgen-1b")
-model = AutoModelForCausalLM.from_pretrained("ddidacus/RS-molgen-1b")
 generation_kwargs = {
     "max_new_tokens": 128,
     "min_length": -1,

 # Setup
 device = "cuda"
+tokenizer = AutoTokenizer.from_pretrained("ddidacus/RS-mol-llama-1b")
+model = AutoModelForCausalLM.from_pretrained("ddidacus/RS-mol-llama-1b")
 generation_kwargs = {
     "max_new_tokens": 128,
     "min_length": -1,