Spaces:
Runtime error
Runtime error
Update app
Browse files
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",
|