Update tts_server.py

This commit is contained in:
erew123 authored and GitHub committed 2024-11-03 12:57:43 +00:00
1 parent 6ba6549f0f
commit 8e466ee119
1 file changed
+4
+4
View File
@@ -36,6 +36,10 @@ from pydantic import BaseModel, ValidationError, Field, field_validator
import warnings
warnings.filterwarnings("ignore", category=UserWarning, module="torch.nn.utils.weight_norm")
warnings.filterwarnings("ignore", category=UserWarning, module="torch.nn.functional", lineno=5476)
# Filter ComplexHalf support warning
warnings.filterwarnings("ignore", message="ComplexHalf support is experimental")
# Filter Flash Attention warning
warnings.filterwarnings("ignore", message="1Torch was not compiled with flash attention"
####################
# Setup local path #
####################