comi-logic/Wan2GP
7
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Vace powercharged

Browse Source
This commit is contained in:
DeepBeepMeep
2025-06-17 23:45:47 +02:00
parent 826cc3adb7
commit febeb95767
18 changed files with 1945 additions and 605 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
hyvideo/hunyuan.py
View File

@@ -315,7 +315,7 @@ class Inference(object):
@classmethod
def from_pretrained(cls, model_filepath, base_model_type, text_encoder_filepath, dtype = torch.bfloat16, VAE_dtype = torch.float16, mixed_precision_transformer =torch.bfloat16 , quantizeTransformer = False, save_quantized = False, **kwargs):
def from_pretrained(cls, model_filepath, model_type, base_model_type, text_encoder_filepath, dtype = torch.bfloat16, VAE_dtype = torch.float16, mixed_precision_transformer =torch.bfloat16 , quantizeTransformer = False, save_quantized = False, **kwargs):
device = "cuda"
@@ -392,8 +392,8 @@ class Inference(object):
# offload.save_model(model, "hunyuan_video_avatar_edit_720_bf16.safetensors")
# offload.save_model(model, "hunyuan_video_avatar_edit_720_quanto_bf16_int8.safetensors", do_quantize= True)
if save_quantized:
from wan.utils.utils import save_quantized_model
save_quantized_model(model, filepath, dtype, None)
from wgp import save_quantized_model
save_quantized_model(model, model_type, filepath, dtype, None)
model.mixed_precision = mixed_precision_transformer