Spaces:

ikot
/

aitools

Running

ikot commited on Oct 2

Commit

b9c0e72

•

1 Parent(s): c3ce0aa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import spaces
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
-model_name = "Qwen/Qwen2.5-3B-Instruct"
 model = AutoModelForCausalLM.from_pretrained(
     model_name,

 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+model_name = "Qwen/Qwen2.5-1.5B-Instruct"
 model = AutoModelForCausalLM.from_pretrained(
     model_name,