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:
|