Spaces:

harkov000
/

peft-lora-sd-dreambooth

Runtime error

harkov000 commited on Feb 22, 2023

Commit

0d6d3ac

1 Parent(s): a1a0fc5

Added dreamlike-photoreal-2.0 support

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,6 +65,7 @@ def create_training_demo(trainer: Trainer, pipe: InferencePipeline) -> gr.Blocks
                 "CompVis/stable-diffusion-v1-4",
                 "runwayml/stable-diffusion-v1-5",
                 "stabilityai/stable-diffusion-2-1-base",
             ],
             value="runwayml/stable-diffusion-v1-5",
             label="Base Model",
@@ -250,6 +251,7 @@ def create_inference_demo(pipe: InferencePipeline) -> gr.Blocks:
                         "CompVis/stable-diffusion-v1-4",
                         "runwayml/stable-diffusion-v1-5",
                         "stabilityai/stable-diffusion-2-1-base",
                     ],
                     value="runwayml/stable-diffusion-v1-5",
                     label="Base Model",

                 "CompVis/stable-diffusion-v1-4",
                 "runwayml/stable-diffusion-v1-5",
                 "stabilityai/stable-diffusion-2-1-base",
+                "dreamlike-art/dreamlike-photoreal-2.0"
             ],
             value="runwayml/stable-diffusion-v1-5",
             label="Base Model",
                         "CompVis/stable-diffusion-v1-4",
                         "runwayml/stable-diffusion-v1-5",
                         "stabilityai/stable-diffusion-2-1-base",
+                        "dreamlike-art/dreamlike-photoreal-2.0"
                     ],
                     value="runwayml/stable-diffusion-v1-5",
                     label="Base Model",