Spaces:

khang119966
/

Vintern-v2-Demo

Runtime error

qnguyen3 commited on Apr 8, 2024

Commit

2e83738

verified ·

1 Parent(s): 88b9346

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,6 @@ tokenizer = AutoTokenizer.from_pretrained(
 model = AutoModelForCausalLM.from_pretrained(
     'qnguyen3/nanoLLaVA',
     torch_dtype=torch.float16,
-    device_map='auto',
     trust_remote_code=True)
 model.to("cuda:0")

 model = AutoModelForCausalLM.from_pretrained(
     'qnguyen3/nanoLLaVA',
     torch_dtype=torch.float16,
     trust_remote_code=True)
 model.to("cuda:0")