Spaces:

joackoEsp
/

Solid-ALIA

Paused

joackoEsp commited on Mar 2

Commit

a9b1392

verified ·

1 Parent(s): 3bd1c2b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 # Cargar el modelo y el tokenizador
-model_name = "BSC-LT/salamandra-7b-instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16)

 import torch
 # Cargar el modelo y el tokenizador
+model_name = "BSC-LT/salamandra-2b"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16)