GiorgioV commited on
Commit
1e52d76
·
verified ·
1 Parent(s): 5ab0a82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -3
app.py CHANGED
@@ -1,8 +1,17 @@
1
  import spaces
2
  import torch
3
- from diffusers.pipelines.wan.pipeline_wan_i2v import WanImageToVideoPipeline
4
- from diffusers.models.transformers.transformer_wan import WanTransformer3DModel
5
- from diffusers.utils.export_utils import export_to_video
 
 
 
 
 
 
 
 
 
6
  import gradio as gr
7
  import tempfile
8
  import numpy as np
@@ -10,6 +19,10 @@ from PIL import Image
10
  import random
11
  import gc
12
 
 
 
 
 
13
  from torchao.quantization import quantize_
14
  from torchao.quantization import Float8DynamicActivationFloat8WeightConfig
15
  from torchao.quantization import Int8WeightOnlyConfig
 
1
  import spaces
2
  import torch
3
+ from transformers import (
4
+ AutoTokenizer,
5
+ AutoImageProcessor,
6
+ T5EncoderModel,
7
+ )
8
+
9
+ from diffusers import (
10
+ WanImageToVideoPipeline,
11
+ WanTransformer3DModel,
12
+ AutoencoderKL,
13
+ EulerDiscreteScheduler,
14
+ )
15
  import gradio as gr
16
  import tempfile
17
  import numpy as np
 
19
  import random
20
  import gc
21
 
22
+ from diffusers.utils.export_utils import export_to_video
23
+ from huggingface_hub import hf_hub_download
24
+ from safetensors.torch import load_file as safetensors_load
25
+
26
  from torchao.quantization import quantize_
27
  from torchao.quantization import Float8DynamicActivationFloat8WeightConfig
28
  from torchao.quantization import Int8WeightOnlyConfig