Xuweiyi commited on
Commit
b72a497
·
verified ·
1 Parent(s): 7f7ae3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -16,6 +16,8 @@ from safetensors import safe_open
16
  from diffusers import AutoencoderKL
17
  from diffusers import DDIMScheduler
18
  from diffusers.utils.import_utils import is_xformers_available
 
 
19
  from transformers import CLIPTextModel, CLIPTokenizer
20
 
21
  from animatediff.models.unet import UNet3DConditionModel
 
16
  from diffusers import AutoencoderKL
17
  from diffusers import DDIMScheduler
18
  from diffusers.utils.import_utils import is_xformers_available
19
+ import transformer
20
+ transformers.utils.move_cache()
21
  from transformers import CLIPTextModel, CLIPTokenizer
22
 
23
  from animatediff.models.unet import UNet3DConditionModel