Hugging Face's logo

Spaces:
qnguyen3
/
nanoLLaVA
Runtime error

App Files Community
3
qnguyen3 commited on
Commit
e6eaffd
·
verified ·
1 Parent(s): d5f6db7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ tokenizer = AutoTokenizer.from_pretrained(
18
  trust_remote_code=True)
19
 
20
  model = LlavaQwen2ForCausalLM.from_pretrained(
21
- 'qnguyen3/nanoLLaVA',
22
  torch_dtype=torch.float16,
23
  attn_implementation="flash_attention_2",
24
  trust_remote_code=True)
 
18
  trust_remote_code=True)
19
 
20
  model = LlavaQwen2ForCausalLM.from_pretrained(
21
+ 'qnguyen3/nanoLLaVA-1.5',
22
  torch_dtype=torch.float16,
23
  attn_implementation="flash_attention_2",
24
  trust_remote_code=True)