-
Notifications
You must be signed in to change notification settings - Fork 72
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
adding hunyuan hf (support lora finetuning); unified hunyuan hf inference with quantization #135
Conversation
This reverts commit 580575c.
initialize_sequence_parallel_state(world_size) | ||
|
||
|
||
def inference(args): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why separate inference and inference quantization functions?
fastvideo/train.py
Outdated
# print("2222222222222222222222222222222222222222222222") | ||
# print(type(latents_attention_mask)) | ||
# print(latents_attention_mask) | ||
training_guidance = torch.tensor([1000.0], |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
training guidance only works with Hunyuan. Adding this would make Mochi mal function,
fastvideo/utils/checkpoint.py
Outdated
@@ -46,6 +45,14 @@ def save_checkpoint_optimizer(model, | |||
"diffusion_pytorch_model.safetensors") | |||
save_file(cpu_state, weight_path) | |||
config_dict = dict(model.config) | |||
config_dict.pop('dtype') | |||
# dtype = config_dict['dtype'] |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
delete commented lines?
@@ -1,19 +1,19 @@ | |||
#!/bin/bash | |||
|
|||
num_gpus=1 | |||
export MODEL_BASE="data/FastHunyuan-diffusers" | |||
export MODEL_BASE="data/FastHunyuan" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should be FastHunyuan-diffusers?
--pretrained_model_name_or_path data/hunyuan_diffusers \ | ||
--model_type hunyuan_hf \ | ||
--cache_dir data/.cache \ | ||
--data_json_path data/Image-Vid-Finetune-HunYuan/videos2caption.json \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This dataset path should be set to the 22 videos?
--cfg 0.0 \ | ||
--ema_decay 0.999 \ | ||
--log_validation \ | ||
--output_dir data/outputs/HSH-Taylor-Finetune-Hunyuan \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
incorrect name.
…ence with quantization (hao-ai-lab#135)
adding hunyuan hf (support lora finetuning); unified hunyuan hf inference with quantization