farrell236 commited on
Commit
078fffd
·
verified ·
1 Parent(s): 180772b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -106,7 +106,7 @@ def generate(
106
  videos=video_inputs,
107
  padding=True,
108
  return_tensors="pt",
109
- ).to("cuda")
110
  streamer = TextIteratorStreamer(processor, skip_special_tokens=True, skip_prompt=True)
111
 
112
  generation_kwargs = dict(
 
106
  videos=video_inputs,
107
  padding=True,
108
  return_tensors="pt",
109
+ ) # .to("cuda")
110
  streamer = TextIteratorStreamer(processor, skip_special_tokens=True, skip_prompt=True)
111
 
112
  generation_kwargs = dict(