Spaces:

doubledsbv
/

gemma_2_2B_kafka_experimental

Sleeping

doubledsbv commited on Apr 24

Commit

2734e1e

•

1 Parent(s): aad79ca

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import transformers
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model_id = "doubledsbv/Llama-3-Kafka-8B-v0.3"
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
 tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model_id = "doubledsbv/Llama-3-Kafka-8B-v0.2"
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
 tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)