ArrcttacsrjksX commited on
Commit
b12bef8
·
verified ·
1 Parent(s): 9c21ea5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
3
  import torch
4
 
5
  # Load model và tokenizer (chọn model mày muốn ở đây)
6
- model_id = "Qwen/Qwen3-0.6B-Chat"
7
 
8
  tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
9
  model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True)
 
3
  import torch
4
 
5
  # Load model và tokenizer (chọn model mày muốn ở đây)
6
+ model_id = "Qwen/Qwen3-0.6B-Base"
7
 
8
  tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
9
  model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True)