Hugging Face's logo

Spaces:
TH9817
/
aa
Runtime error

App Files Community
TH9817 commited on
Commit
ef4a010
·
verified ·
1 Parent(s): 80b2511

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ model = LlavaNextVideoForConditionalGeneration.from_pretrained(
15
  model_id,
16
  torch_dtype=torch.float16,
17
  low_cpu_mem_usage=True,
18
- )
19
 
20
  processor = LlavaNextVideoProcessor.from_pretrained(model_id)
21
 
 
15
  model_id,
16
  torch_dtype=torch.float16,
17
  low_cpu_mem_usage=True,
18
+ ).to(0)
19
 
20
  processor = LlavaNextVideoProcessor.from_pretrained(model_id)
21