Update model_multitalk.py

This commit is contained in:
kijai 2025-10-03 18:27:29 +03:00
parent 6f6db12bbe
commit 00c069dd1c

View File

@ -591,4 +591,4 @@ class InfiniteTalkOuterSampleLoopingWrapper:
self.encoded_audio = [ea.to(device_or_dtype) for ea in self.encoded_audio]
if self.ref_target_masks is not None:
self.ref_target_masks = self.ref_target_masks.to(device_or_dtype)
return self
return self