srijansengupta commited on
Commit
79c2018
·
verified ·
1 Parent(s): 2268486

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -4,8 +4,7 @@ import os
4
  model = AutoModelForCausalLM.from_pretrained(
5
  "Qwen/CodeQwen1.5-7B-Chat",
6
  torch_dtype="auto",
7
- device_map="auto",
8
- token=hf_token
9
  )
10
  tokenizer = AutoTokenizer.from_pretrained("Qwen/CodeQwen1.5-7B-Chat")
11
 
 
4
  model = AutoModelForCausalLM.from_pretrained(
5
  "Qwen/CodeQwen1.5-7B-Chat",
6
  torch_dtype="auto",
7
+ device_map="auto"
 
8
  )
9
  tokenizer = AutoTokenizer.from_pretrained("Qwen/CodeQwen1.5-7B-Chat")
10