ghunkins commited on
Commit
17add40
·
verified ·
1 Parent(s): d119cf2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -0
README.md CHANGED
@@ -33,6 +33,16 @@ from hyvideo.diffusion.pipelines.pipeline_hunyuan_video import HunyuanVideoPipel
33
  from hyvideo.modules.models import HYVideoDiffusionTransformer
34
  from hyvideo.vae.autoencoder_kl_causal_3d import AutoencoderKLCausal3D
35
  import diffusers.pipelines
 
 
 
 
 
 
 
 
 
 
36
  self.set_nested_attr(diffusers.pipelines, 'hyvideo.HunyuanVideoPipeline', HunyuanVideoPipeline)
37
  self.set_nested_attr(diffusers.pipelines, 'hyvideo.HYVideoDiffusionTransformer', HYVideoDiffusionTransformer)
38
  self.set_nested_attr(diffusers.pipelines, 'hyvideo.AutoencoderKLCausal3D', AutoencoderKLCausal3D)
 
33
  from hyvideo.modules.models import HYVideoDiffusionTransformer
34
  from hyvideo.vae.autoencoder_kl_causal_3d import AutoencoderKLCausal3D
35
  import diffusers.pipelines
36
+
37
+ from types import ModuleType
38
+ def set_nested_attr(current, path, value):
39
+ parts = path.split('.')
40
+ for part in parts[:-1]:
41
+ if not hasattr(current, part):
42
+ setattr(current, part, ModuleType(part))
43
+ current = getattr(current, part)
44
+ setattr(current, parts[-1], value)
45
+
46
  self.set_nested_attr(diffusers.pipelines, 'hyvideo.HunyuanVideoPipeline', HunyuanVideoPipeline)
47
  self.set_nested_attr(diffusers.pipelines, 'hyvideo.HYVideoDiffusionTransformer', HYVideoDiffusionTransformer)
48
  self.set_nested_attr(diffusers.pipelines, 'hyvideo.AutoencoderKLCausal3D', AutoencoderKLCausal3D)