Fabrice-TIERCELIN commited on
Commit
0575a63
·
verified ·
1 Parent(s): 95001cb
Files changed (1) hide show
  1. app.py +3 -0
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()