multimodalart HF Staff commited on
Commit
8ed752e
·
verified ·
1 Parent(s): bfecffd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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):