From 8fd62b3e78aa28ebb926f862daa8b2f69c2ecae1 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 14 Jul 2025 09:13:41 +0300
Subject: [PATCH] Multitalk wav2vec model doesn't actually need the .pt file

chinese-wav2vec2-base-fairseq-ckpt.pt can be safely deleted and won't be autodownloaded in the future
---
 fantasytalking/nodes.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/fantasytalking/nodes.py b/fantasytalking/nodes.py
index 419db6b..2458407 100644
--- a/fantasytalking/nodes.py
+++ b/fantasytalking/nodes.py
@@ -56,6 +56,8 @@ class DownloadAndLoadWav2VecModel:
             ignore_patterns = None
             if model == "facebook/wav2vec2-base-960h":
                 ignore_patterns = ["*.bin", "*.h5"]
+            elif model == "TencentGameMate/chinese-wav2vec2-base":
+                ignore_patterns = ["*.pt"]
             snapshot_download(
                 repo_id=model,
                 ignore_patterns=ignore_patterns,