Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ from diffusers.models.attention_processor import AttnProcessor2_0
|
|
24 |
from transformers import CLIPTextModel, CLIPTokenizer
|
25 |
from diffusers_vdm.pipeline import LatentVideoDiffusionPipeline
|
26 |
from diffusers_vdm.utils import resize_and_center_crop, save_bcthw_as_mp4
|
27 |
-
|
28 |
@spaces.GPU()
|
29 |
class ModifiedUNet(UNet2DConditionModel):
|
30 |
@classmethod
|
|
|
24 |
from transformers import CLIPTextModel, CLIPTokenizer
|
25 |
from diffusers_vdm.pipeline import LatentVideoDiffusionPipeline
|
26 |
from diffusers_vdm.utils import resize_and_center_crop, save_bcthw_as_mp4
|
27 |
+
torch.set_grad_enabled(False)
|
28 |
@spaces.GPU()
|
29 |
class ModifiedUNet(UNet2DConditionModel):
|
30 |
@classmethod
|