Spaces:

C2MV
/

Biotech2

Running on Zero

C2MV commited on Sep 23

Commit

87dafbb

•

1 Parent(s): 4ba28e5

Update interface.py

Files changed (1) hide show

interface.py CHANGED Viewed

@@ -17,7 +17,7 @@ import json
 # Inicialización del dispositivo
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model_path = "Qwen/Qwen2.5-32B-Instruct" #"01-ai/Yi-Coder-9B-Chat"
 # Carga del tokenizer y modelo
 tokenizer = AutoTokenizer.from_pretrained(model_path)

 # Inicialización del dispositivo
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model_path = "Qwen/Qwen2.5-14B-Instruct" #"01-ai/Yi-Coder-9B-Chat"
 # Carga del tokenizer y modelo
 tokenizer = AutoTokenizer.from_pretrained(model_path)