anthony01 commited on
Commit
0bb5e70
·
1 Parent(s): d9d8ec3

fix: typo in interface.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -127,7 +127,7 @@ def load_model(
127
  checkpoint_path = os.path.join("/tmp", model_name)
128
 
129
  # 3. Download it to /tmp if it isn't already there
130
- if not os.path.isfile(tmp_checkpoint_path):
131
  link = pretrained_links.get(model_name)
132
  if not link or link == "link":
133
  raise FileNotFoundError(f"No pretrained link for {model_name}")
 
127
  checkpoint_path = os.path.join("/tmp", model_name)
128
 
129
  # 3. Download it to /tmp if it isn't already there
130
+ if not os.path.isfile(checkpoint_path):
131
  link = pretrained_links.get(model_name)
132
  if not link or link == "link":
133
  raise FileNotFoundError(f"No pretrained link for {model_name}")