John6666 commited on
Commit
99f0ea5
·
verified ·
1 Parent(s): 777e8ec

Upload 2 files

Browse files
Files changed (1) hide show
  1. convert_url_to_diffusers_sd_gr.py +3 -3
convert_url_to_diffusers_sd_gr.py CHANGED
@@ -310,16 +310,16 @@ def convert_url_to_diffusers_repo(dl_url, new_repo_id, hf_token, civitai_key="",
310
  if not is_repo_name(new_repo_id):
311
  print(f"Invalid repo name: {new_repo_id}")
312
  progress(1, desc=f"Invalid repo name: {new_repo_id}")
313
- return ""
314
  if is_repo_exists(new_repo_id):
315
  print(f"Repo already exists: {new_repo_id}")
316
  progress(1, desc=f"Repo already exists: {new_repo_id}")
317
- return ""
318
  lora_dict = {lora1: lora1s, lora2: lora2s, lora3: lora3s, lora4: lora4s, lora5: lora5s}
319
  if None in lora_dict.keys(): del lora_dict[None]
320
  new_path = convert_url_to_diffusers_sd(dl_url, civitai_key, half, vae, scheduler, lora_dict,
321
  model_type, sample_size, ema)
322
- if not new_path: return ""
323
  repo_url = create_diffusers_repo(new_repo_id, new_path)
324
  if not repo_urls: repo_urls = []
325
  repo_urls.append(repo_url)
 
310
  if not is_repo_name(new_repo_id):
311
  print(f"Invalid repo name: {new_repo_id}")
312
  progress(1, desc=f"Invalid repo name: {new_repo_id}")
313
+ return gr.update(value=repo_urls, choices=repo_urls), gr.update(value="")
314
  if is_repo_exists(new_repo_id):
315
  print(f"Repo already exists: {new_repo_id}")
316
  progress(1, desc=f"Repo already exists: {new_repo_id}")
317
+ return gr.update(value=repo_urls, choices=repo_urls), gr.update(value="")
318
  lora_dict = {lora1: lora1s, lora2: lora2s, lora3: lora3s, lora4: lora4s, lora5: lora5s}
319
  if None in lora_dict.keys(): del lora_dict[None]
320
  new_path = convert_url_to_diffusers_sd(dl_url, civitai_key, half, vae, scheduler, lora_dict,
321
  model_type, sample_size, ema)
322
+ if not new_path: return gr.update(value=repo_urls, choices=repo_urls), gr.update(value="")
323
  repo_url = create_diffusers_repo(new_repo_id, new_path)
324
  if not repo_urls: repo_urls = []
325
  repo_urls.append(repo_url)