little change
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user