ddidacus
/

Mol-MoE-6x1b

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

ddidacus commited on Feb 7

Commit

094f8e5

·

verified ·

1 Parent(s): 1eb3cd1

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -22,8 +22,8 @@ import re
 # Setup
 device = "cuda"
-tokenizer = AutoTokenizer.from_pretrained("ddidacus/Mol-MoE-5x1b")
-model = AutoModelForCausalLM.from_pretrained("ddidacus/Mol-MoE-5x1b")
 generation_kwargs = {
     "max_new_tokens": 128,
     "min_length": -1,

 # Setup
 device = "cuda"
+tokenizer = AutoTokenizer.from_pretrained("ddidacus/Mol-MoE-6x1b")
+model = AutoModelForCausalLM.from_pretrained("ddidacus/Mol-MoE-6x1b")
 generation_kwargs = {
     "max_new_tokens": 128,
     "min_length": -1,