Spaces:
Running
Running
import gc
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
from diffusers_helper.hf_login import login
|
2 |
|
|
|
3 |
import os
|
4 |
|
5 |
os.environ['HF_HOME'] = os.path.abspath(os.path.realpath(os.path.join(os.path.dirname(__file__), './hf_download')))
|
@@ -46,6 +47,8 @@ from transformers import SiglipImageProcessor, SiglipVisionModel
|
|
46 |
from diffusers_helper.clip_vision import hf_clip_vision_encode
|
47 |
from diffusers_helper.bucket_tools import find_nearest_bucket
|
48 |
from diffusers import BitsAndBytesConfig as DiffusersBitsAndBytesConfig, HunyuanVideoTransformer3DModel, HunyuanVideoPipeline
|
|
|
|
|
49 |
import pillow_heif
|
50 |
|
51 |
pillow_heif.register_heif_opener()
|
|
|
1 |
from diffusers_helper.hf_login import login
|
2 |
|
3 |
+
import gc
|
4 |
import os
|
5 |
|
6 |
os.environ['HF_HOME'] = os.path.abspath(os.path.realpath(os.path.join(os.path.dirname(__file__), './hf_download')))
|
|
|
47 |
from diffusers_helper.clip_vision import hf_clip_vision_encode
|
48 |
from diffusers_helper.bucket_tools import find_nearest_bucket
|
49 |
from diffusers import BitsAndBytesConfig as DiffusersBitsAndBytesConfig, HunyuanVideoTransformer3DModel, HunyuanVideoPipeline
|
50 |
+
from utils.lora_utils import merge_lora_to_state_dict
|
51 |
+
from utils.fp8_optimization_utils import optimize_state_dict_with_fp8, apply_fp8_monkey_patch
|
52 |
import pillow_heif
|
53 |
|
54 |
pillow_heif.register_heif_opener()
|