Spaces:

darkstorm2150
/

protogen-web-ui

Running on A10G

darkstorm2150 commited on May 22, 2023

Commit

f02577d

•

1 Parent(s): 90f9f38

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -45,7 +45,7 @@ def get_model(name):
     model = MODELS[name]
     if model.pipe_t2i is None:
-        model.pip_t2i = StableDiffusionPipeline.from_pretrained(
             model.path, torch_dtype=torch.float16, safety_checker=SAFETY_CHECKER
         )
         model.pipe_t2i.scheduler = DPMSolverMultistepScheduler.from_config(

     model = MODELS[name]
     if model.pipe_t2i is None:
+        model.pipe_t2i = StableDiffusionPipeline.from_pretrained(
             model.path, torch_dtype=torch.float16, safety_checker=SAFETY_CHECKER
         )
         model.pipe_t2i.scheduler = DPMSolverMultistepScheduler.from_config(