little change

This commit is contained in:
songkey-win
2025-06-25 23:50:43 +08:00
parent 47fe61445e
commit f7ece3f7d4
6 changed files with 521 additions and 661 deletions
-4
View File
@@ -75,12 +75,8 @@ class HM5VideoPipeline(HMPipeline):
else:
hm_adapter = HM5ReferenceAdapter.from_pretrained(hm_reference_dir)
# hm_adapter = HM5ReferenceAdapter.from_pretrained(hm_reference_dir)
motion_adapter = HM5MotionAdapter.from_pretrained(hm_motion_dir)
stats = torch.load(r"E:\upload2\weights\a100-0617mpsd15vff\checkpoint-41000.pth", map_location='cpu')
motion_adapter.load_state_dict(stats['motion'], strict=False)
if isinstance(self.unet, HM3Denoising3D):
self.unet.insert_reference_adapter(hm_adapter)
self.unet.insert_reference_adapter(motion_adapter)