Spaces:

quancute
/

new_demo

Sleeping

quancute commited on Oct 22, 2024

Commit

65af8d4

verified ·

1 Parent(s): a1a5866

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ print('device =', device)
 # Load mô hình và tokenizer
 model = AutoModelForCausalLM.from_pretrained(
-    'Viet-Mistral/Vistral-7B-Chat',
     torch_dtype=torch.bfloat16, # change to torch.float16 if you're using V100
     device_map="auto"
 )

 # Load mô hình và tokenizer
 model = AutoModelForCausalLM.from_pretrained(
+     MODEL,
     torch_dtype=torch.bfloat16, # change to torch.float16 if you're using V100
     device_map="auto"
 )