Spaces:

fschwartzer
/

llm_test

Sleeping

fschwartzer commited on Jul 3

Commit

2a28303

•

1 Parent(s): 6685c0c

Update src/brain.py

Files changed (1) hide show

src/brain.py CHANGED Viewed

@@ -1,6 +1,6 @@
 from transformers import AutoTokenizer, AutoModelForCausalLM
-model_name = "google/gemma-2bgoogle/gemma-2-9b-it"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)

 from transformers import AutoTokenizer, AutoModelForCausalLM
+model_name = "google/gemma-2b"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)