Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -96,7 +96,7 @@ Set the environment variable `model` to change the model:
|
|
96 |
'liuhaotian/llava-v1.6-vicuna-7b',
|
97 |
]
|
98 |
bits = int(os.getenv("bits", 16))
|
99 |
-
model = os.getenv("model", models[
|
100 |
available_devices = os.getenv("CUDA_VISIBLE_DEVICES", "0")
|
101 |
model_path, model_name = model, model.split("/")[-1]
|
102 |
|
@@ -104,7 +104,7 @@ Set the environment variable `model` to change the model:
|
|
104 |
|
105 |
|
106 |
# Wait for worker and controller to start
|
107 |
-
time.sleep(
|
108 |
|
109 |
exit_status = 0
|
110 |
try:
|
|
|
96 |
'liuhaotian/llava-v1.6-vicuna-7b',
|
97 |
]
|
98 |
bits = int(os.getenv("bits", 16))
|
99 |
+
model = os.getenv("model", models[0])
|
100 |
available_devices = os.getenv("CUDA_VISIBLE_DEVICES", "0")
|
101 |
model_path, model_name = model, model.split("/")[-1]
|
102 |
|
|
|
104 |
|
105 |
|
106 |
# Wait for worker and controller to start
|
107 |
+
time.sleep(60)
|
108 |
|
109 |
exit_status = 0
|
110 |
try:
|