Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -86,9 +86,9 @@ import gradio as gr
|
|
| 86 |
# --- Part 3: Model Initialization (Pre-Load) ---
|
| 87 |
|
| 88 |
# Initialize Distributed/Infer States
|
| 89 |
-
parallel_dims = initialize_parallel_state(sp=1)
|
| 90 |
-
if torch.cuda.is_available():
|
| 91 |
-
torch.cuda.set_device(0)
|
| 92 |
|
| 93 |
class ArgsNamespace:
|
| 94 |
def __init__(self):
|
|
|
|
| 86 |
# --- Part 3: Model Initialization (Pre-Load) ---
|
| 87 |
|
| 88 |
# Initialize Distributed/Infer States
|
| 89 |
+
#parallel_dims = initialize_parallel_state(sp=1)
|
| 90 |
+
#if torch.cuda.is_available():
|
| 91 |
+
# torch.cuda.set_device(0)
|
| 92 |
|
| 93 |
class ArgsNamespace:
|
| 94 |
def __init__(self):
|