Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ import torch
|
|
18 |
import deepspeed # Import DeepSpeed for ZeroGPU
|
19 |
|
20 |
# Suppress warnings
|
21 |
-
warnings.filterwarnings('ignore', category='
|
22 |
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
|
23 |
|
24 |
def initialize_zero_gpu():
|
|
|
18 |
import deepspeed # Import DeepSpeed for ZeroGPU
|
19 |
|
20 |
# Suppress warnings
|
21 |
+
warnings.filterwarnings('ignore', category='UserWarning')
|
22 |
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
|
23 |
|
24 |
def initialize_zero_gpu():
|