Spaces:
Paused
Paused
Update app.py
Browse files
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(
|