Approximetal commited on
Commit
3d2ae33
·
verified ·
1 Parent(s): 5f1dd8a

Update inference_gradio.py

Browse files
Files changed (1) hide show
  1. 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
- else:
147
- # No local ckpts for this project, try remote mapping
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)