Hugging Face's logo

Spaces:
Michael4933
/
Migician
Runtime error

App Files Community
Michael4933 commited on
Commit
c4f6597
·
verified ·
1 Parent(s): d33c8f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -17,10 +17,7 @@ Come and feel the magic of multi-image grounding!
17
 
18
  model_id = "Michael4933/Migician"
19
  model = Qwen2VLForConditionalGeneration.from_pretrained(
20
- model_id,
21
- torch_dtype=torch.bfloat16,
22
- attn_implementation="flash_attention_2",
23
- device_map="auto",
24
  )
25
  processor = AutoProcessor.from_pretrained(model_id)
26
 
 
17
 
18
  model_id = "Michael4933/Migician"
19
  model = Qwen2VLForConditionalGeneration.from_pretrained(
20
+ model_id, torch_dtype="auto", device_map="auto"
 
 
 
21
  )
22
  processor = AutoProcessor.from_pretrained(model_id)
23