r3gm commited on
Commit
2ee02e1
1 Parent(s): 897baeb

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -300,7 +300,7 @@ def get_model_type(repo_id: str):
300
  default = "SD 1.5"
301
  try:
302
  if os.path.exists(repo_id):
303
- tag = checkpoint_model_type(repo_id)
304
  return DIFFUSECRAFT_CHECKPOINT_NAME[tag]
305
  else:
306
  model = api.model_info(repo_id=repo_id, timeout=5.0)
 
300
  default = "SD 1.5"
301
  try:
302
  if os.path.exists(repo_id):
303
+ tag, _, _ = checkpoint_model_type(repo_id)
304
  return DIFFUSECRAFT_CHECKPOINT_NAME[tag]
305
  else:
306
  model = api.model_info(repo_id=repo_id, timeout=5.0)