Hugging Face's logo

Spaces:
fffiloni
/
MEMO
Running on Zero

App Files Community
5
fffiloni commited on
Commit
385d944
·
verified ·
1 Parent(s): 5d0a4ce

Update hf_gradio_app.py

Browse files
Files changed (1) hide show
  1. hf_gradio_app.py +2 -2
hf_gradio_app.py CHANGED
@@ -76,8 +76,8 @@ reference_net.requires_grad_(False).eval()
76
  diffusion_net.requires_grad_(False).eval()
77
  image_proj.requires_grad_(False).eval()
78
  audio_proj.requires_grad_(False).eval()
79
- reference_net.enable_xformers_memory_efficient_attention()
80
- diffusion_net.enable_xformers_memory_efficient_attention()
81
  noise_scheduler = FlowMatchEulerDiscreteScheduler()
82
  pipeline = VideoPipeline(vae=vae, reference_net=reference_net, diffusion_net=diffusion_net, scheduler=noise_scheduler, image_proj=image_proj)
83
  #pipeline.to(device=device, dtype=weight_dtype)
 
76
  diffusion_net.requires_grad_(False).eval()
77
  image_proj.requires_grad_(False).eval()
78
  audio_proj.requires_grad_(False).eval()
79
+ #reference_net.enable_xformers_memory_efficient_attention()
80
+ #diffusion_net.enable_xformers_memory_efficient_attention()
81
  noise_scheduler = FlowMatchEulerDiscreteScheduler()
82
  pipeline = VideoPipeline(vae=vae, reference_net=reference_net, diffusion_net=diffusion_net, scheduler=noise_scheduler, image_proj=image_proj)
83
  #pipeline.to(device=device, dtype=weight_dtype)