Hugging Face's logo

Spaces:
qnguyen3
/
nanoLLaVA
Running on Zero

App Files Community
3
qnguyen3 commited on
Commit
2e83738
•
1 Parent(s): 88b9346

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -19,7 +19,6 @@ tokenizer = AutoTokenizer.from_pretrained(
19
  model = AutoModelForCausalLM.from_pretrained(
20
  'qnguyen3/nanoLLaVA',
21
  torch_dtype=torch.float16,
22
- device_map='auto',
23
  trust_remote_code=True)
24
 
25
  model.to("cuda:0")
 
19
  model = AutoModelForCausalLM.from_pretrained(
20
  'qnguyen3/nanoLLaVA',
21
  torch_dtype=torch.float16,
 
22
  trust_remote_code=True)
23
 
24
  model.to("cuda:0")