jiuface commited on
Commit
bcd30aa
·
verified ·
1 Parent(s): 1b3f790

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -1,4 +1,9 @@
 
1
  import os
 
 
 
 
2
  import torch
3
 
4
  from diffusers.pipelines.wan.pipeline_wan_i2v import WanImageToVideoPipeline
@@ -9,7 +14,6 @@ from diffusers.utils import export_to_video
9
  from transformers import CLIPVisionModel
10
  import gradio as gr
11
  import tempfile
12
- import spaces
13
  from huggingface_hub import hf_hub_download
14
  import numpy as np
15
  from PIL import Image
 
1
+ # PyTorch 2.8 (temporary hack)
2
  import os
3
+ os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9" spaces')
4
+
5
+ # Actual demo code
6
+ import spaces
7
  import torch
8
 
9
  from diffusers.pipelines.wan.pipeline_wan_i2v import WanImageToVideoPipeline
 
14
  from transformers import CLIPVisionModel
15
  import gradio as gr
16
  import tempfile
 
17
  from huggingface_hub import hf_hub_download
18
  import numpy as np
19
  from PIL import Image