Spaces:

tsqn
/

Latte-1-low_vram

Runtime error

tsqn commited on Nov 17, 2024

Commit

b9a7927

1 Parent(s): a45ef62

chore: ZeroGPU

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import gc
 import os
 import torch
 import gradio as gr
 from diffusers import LattePipeline
 from transformers import T5EncoderModel, BitsAndBytesConfig
@@ -32,6 +33,7 @@ def initialize_pipeline():
     )
     return pipe, text_encoder
 def generate_video(
     prompt: str,
     negative_prompt: str = "",

 import gc
 import os
 import torch
+import spaces
 import gradio as gr
 from diffusers import LattePipeline
 from transformers import T5EncoderModel, BitsAndBytesConfig
     )
     return pipe, text_encoder
+@spaces.GPU(duration=120)
 def generate_video(
     prompt: str,
     negative_prompt: str = "",