update
Browse files
model.py
CHANGED
|
@@ -59,8 +59,8 @@ class INF5Model(PreTrainedModel):
|
|
| 59 |
)
|
| 60 |
)
|
| 61 |
|
| 62 |
-
#
|
| 63 |
-
|
| 64 |
|
| 65 |
|
| 66 |
def forward(self, text: str, ref_audio_path: str, ref_text: str):
|
|
|
|
| 59 |
)
|
| 60 |
)
|
| 61 |
|
| 62 |
+
# Load state dict into model
|
| 63 |
+
self.ema_model.load_state_dict(state_dict, strict=False, assign=True)
|
| 64 |
|
| 65 |
|
| 66 |
def forward(self, text: str, ref_audio_path: str, ref_text: str):
|