Skip to content

Commit af6563b

Browse files
committed
add warniong suppression
1 parent f78d1b8 commit af6563b

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

generate.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010

1111
import os
1212
import re
13+
import warnings
1314
from typing import List, Optional
1415

1516
import click
@@ -22,6 +23,9 @@
2223
import legacy
2324
from age_estimator import AgeEstimator
2425

26+
# Suppress CUDA kernel compilation warnings
27+
warnings.filterwarnings('ignore', message='Failed to build CUDA kernels for upfirdn2d')
28+
2529
# ----------------------------------------------------------------------------
2630

2731

0 commit comments

Comments
 (0)