rphrp1985 commited on
Commit
3c97666
·
verified ·
1 Parent(s): 09b1598

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,14 +47,14 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
47
  # model_id = "mistralai/Mistral-7B-v0.3"
48
 
49
  model_id = "01-ai/Yi-1.5-9B-Chat"
50
- # model_id = "Qwen/Qwen2-7B-Instruct"
51
 
52
 
53
  tokenizer = AutoTokenizer.from_pretrained(
54
  # model_id
55
  model_id,
56
  # use_fast=False
57
- , token= token,)
58
 
59
 
60
  accelerator = Accelerator()
 
47
  # model_id = "mistralai/Mistral-7B-v0.3"
48
 
49
  model_id = "01-ai/Yi-1.5-9B-Chat"
50
+ model_id = "Qwen/Qwen2-7B-Instruct"
51
 
52
 
53
  tokenizer = AutoTokenizer.from_pretrained(
54
  # model_id
55
  model_id,
56
  # use_fast=False
57
+ token= token,)
58
 
59
 
60
  accelerator = Accelerator()