wwen1997 commited on
Commit
a183402
1 Parent(s): 3a598bb

Update app.py

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