kadirnar commited on
Commit
b4f278b
·
verified ·
1 Parent(s): 1c7a35b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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