24L

PlayVoice · Jul 11, 2023 · 4024615 · 4024615
1 parent 541737e
commit 4024615
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/prepare/preprocess_ppg.py b/prepare/preprocess_ppg.py
@@ -15,9 +15,9 @@ def load_model(path) -> Whisper:
     print(dims)
     model = Whisper(dims)
     del model.decoder
-    # cut = len(model.encoder.blocks) // 2
-    # cut = -1 * cut
-    # del model.encoder.blocks[cut:]
+    cut = len(model.encoder.blocks) // 4
+    cut = -1 * cut
+    del model.encoder.blocks[cut:]
     model.load_state_dict(checkpoint["model_state_dict"], strict=False)
     model.eval()
     model.half()