linoyts HF Staff commited on
Commit
97ff6b8
·
verified ·
1 Parent(s): 349c4aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ os.system('pip install --upgrade --pre --extra-index-url https://download.pytorc
5
  # Actual demo code
6
  import spaces
7
  import torch
8
- from diffusers import WanPipeline, AutoencoderKLWan
9
  from diffusers.models.transformers.transformer_wan import WanTransformer3DModel
10
  from diffusers.utils.export_utils import export_to_video
11
  import gradio as gr
 
5
  # Actual demo code
6
  import spaces
7
  import torch
8
+ from diffusers import WanPipeline, AutoencoderKLWan, UniPCMultistepScheduler
9
  from diffusers.models.transformers.transformer_wan import WanTransformer3DModel
10
  from diffusers.utils.export_utils import export_to_video
11
  import gradio as gr