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",
|