lnyan commited on
Commit
b0c2192
1 Parent(s): c9b1627

Update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -671,7 +671,7 @@ def get_model(token="", model_choice="", model_path=""):
671
  print(
672
  f"Note that {ModelChoice.INPAINTING_IMG2IMG.value} only support remote model and requires larger vRAM"
673
  )
674
- tmp = StableDiffusion(token=token, inpainting_model=True)
675
  else:
676
  if len(model_name) < 1:
677
  model_name = (
@@ -775,7 +775,7 @@ proceed_button_js = load_js("proceed")
775
  setup_button_js = load_js("setup")
776
 
777
  if RUN_IN_SPACE:
778
- get_model(token=os.environ.get("hftoken", ""), model_choice=ModelChoice.INPAINTING_IMG2IMG)
779
 
780
  blocks = gr.Blocks(
781
  title="StableDiffusion-Infinity",
 
671
  print(
672
  f"Note that {ModelChoice.INPAINTING_IMG2IMG.value} only support remote model and requires larger vRAM"
673
  )
674
+ tmp = StableDiffusion(token=token, model_name="runwayml/stable-diffusion-v1-5", inpainting_model=True)
675
  else:
676
  if len(model_name) < 1:
677
  model_name = (
 
775
  setup_button_js = load_js("setup")
776
 
777
  if RUN_IN_SPACE:
778
+ get_model(token=os.environ.get("hftoken", ""), model_choice=ModelChoice.INPAINTING_IMG2IMG.value)
779
 
780
  blocks = gr.Blocks(
781
  title="StableDiffusion-Infinity",