wwen1997 commited on
Commit
14beebd
1 Parent(s): 771c278

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -369,14 +369,14 @@ class Drag:
369
  )
370
  controlnet = controlnet.to(dtype)
371
 
372
- if is_xformers_available():
373
- import xformers
374
- xformers_version = version.parse(xformers.__version__)
375
- unet.enable_xformers_memory_efficient_attention()
376
- # controlnet.enable_xformers_memory_efficient_attention()
377
- else:
378
- raise ValueError(
379
- "xformers is not available. Make sure it is installed correctly")
380
 
381
  pipe = StableVideoDiffusionInterpControlPipeline.from_pretrained(
382
  "checkpoints/stable-video-diffusion-img2vid-xt",
 
369
  )
370
  controlnet = controlnet.to(dtype)
371
 
372
+ # if is_xformers_available():
373
+ # import xformers
374
+ # xformers_version = version.parse(xformers.__version__)
375
+ # unet.enable_xformers_memory_efficient_attention()
376
+ # # controlnet.enable_xformers_memory_efficient_attention()
377
+ # else:
378
+ # raise ValueError(
379
+ # "xformers is not available. Make sure it is installed correctly")
380
 
381
  pipe = StableVideoDiffusionInterpControlPipeline.from_pretrained(
382
  "checkpoints/stable-video-diffusion-img2vid-xt",