Spaces:
Running
on
Zero
Running
on
Zero
Update inference_gradio.py
Browse files- inference_gradio.py +3 -2
inference_gradio.py
CHANGED
|
@@ -140,11 +140,12 @@ def get_checkpoints_project(project_name=None, is_gradio=True):
|
|
| 140 |
if not files_checkpoints:
|
| 141 |
files_checkpoints = list(remote_ckpts.values())
|
| 142 |
else:
|
|
|
|
| 143 |
if os.path.isdir(checkpoint_dir[0]):
|
| 144 |
files_checkpoints = glob(os.path.join(checkpoint_dir[0], project_name, "*.pt"))
|
| 145 |
files_checkpoints.extend(glob(os.path.join(checkpoint_dir[0], project_name, "*.safetensors")))
|
| 146 |
-
|
| 147 |
-
|
| 148 |
ckpt = remote_ckpts.get(project_name)
|
| 149 |
files_checkpoints = [ckpt] if ckpt is not None else []
|
| 150 |
print("files_checkpoints:", project_name, files_checkpoints)
|
|
|
|
| 140 |
if not files_checkpoints:
|
| 141 |
files_checkpoints = list(remote_ckpts.values())
|
| 142 |
else:
|
| 143 |
+
files_checkpoints = []
|
| 144 |
if os.path.isdir(checkpoint_dir[0]):
|
| 145 |
files_checkpoints = glob(os.path.join(checkpoint_dir[0], project_name, "*.pt"))
|
| 146 |
files_checkpoints.extend(glob(os.path.join(checkpoint_dir[0], project_name, "*.safetensors")))
|
| 147 |
+
# If no local ckpts for this project, try remote mapping
|
| 148 |
+
if not files_checkpoints:
|
| 149 |
ckpt = remote_ckpts.get(project_name)
|
| 150 |
files_checkpoints = [ckpt] if ckpt is not None else []
|
| 151 |
print("files_checkpoints:", project_name, files_checkpoints)
|