Shivlal6660 commited on
Commit
3b5d99b
·
verified ·
1 Parent(s): 23d4d76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
4
  from threading import Thread
5
 
6
  # Smart & Stable 1.5B Model - Identity Locked
7
- model_id = "Qwen/Qwen2.5-1.5B-Instruct"
8
  tokenizer = AutoTokenizer.from_pretrained(model_id)
9
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
10
 
 
4
  from threading import Thread
5
 
6
  # Smart & Stable 1.5B Model - Identity Locked
7
+ model_id = "Qwen/Qwen2.5-0.5B-Instruct"
8
  tokenizer = AutoTokenizer.from_pretrained(model_id)
9
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
10