fix hunyuan vidoe 1.5 weight-load bug

[helloyongyang]

No description provided.

[gemini-code-assist]

Code Review

This pull request updates the checkpoint loading logic in base_model.py to support specific directory structures for hunyuan_video_1.5 and initializes sequence parallel groups in pre_infer.py. Review feedback highlights potential KeyError and AttributeError risks when accessing configuration values directly and recommends using safer access patterns with .get() and null checks to ensure stability across different environments.

[gemini-code-assist]

This block has two potential issues:

1. self.config["seq_parallel"] will raise a KeyError if the key is missing from the configuration.
2. self.config.get("device_mesh").get_group(...) will raise an AttributeError if device_mesh is not present in the config, as get() returns None by default, and None has no attribute get_group.

Using a safer access pattern ensures the code doesn't crash in distributed environments with incomplete configurations.

Suggested change

	if self.config["seq_parallel"]:
	self.seq_p_group = self.config.get("device_mesh").get_group(mesh_dim="seq_p")
	else:
	self.seq_p_group = None
	if self.config.get("seq_parallel"):
	device_mesh = self.config.get("device_mesh")
	self.seq_p_group = device_mesh.get_group(mesh_dim="seq_p") if device_mesh else None
	else:
	self.seq_p_group = None

[gemini-code-assist]

Using self.config["model_cls"] can raise a KeyError if the key is missing from the configuration. Since BaseTransformerModel is a base class used by multiple model types (such as z_image or qwen_image), this change could cause regressions for other models that do not define model_cls in their config. It is safer to use .get() to check the model class.

Suggested change

	if self.config["model_cls"] == "hunyuan_video_1.5":
	safetensors_files = glob.glob(os.path.join(safetensors_path, "transformer", self.config["transformer_model_name"], "*.safetensors"))
	else:
	safetensors_files = glob.glob(os.path.join(safetensors_path, "*.safetensors"))
	if self.config.get("model_cls") == "hunyuan_video_1.5":
	safetensors_files = glob.glob(os.path.join(safetensors_path, "transformer", self.config.get("transformer_model_name", ""), "*.safetensors"))
	else:
	safetensors_files = glob.glob(os.path.join(safetensors_path, "*.safetensors"))