fix

2023-01-12 20:48:57 +01:00
parent dd9165a99e
commit baaa6ef18a
1 changed files with 2 additions and 2 deletions
--- a/gradio_ui.py
+++ b/gradio_ui.py
@@ -253,9 +253,9 @@ class BlendingFrontend():
 if __name__ == "__main__":    
    
    fp_ckpt = "../stable_diffusion_models/ckpt/v2-1_512-ema-pruned.ckpt" 
-    # sdh = StableDiffusionHolder(fp_ckpt)
+    sdh = StableDiffusionHolder(fp_ckpt)
    
-    self = BlendingFrontend(None)
+    self = BlendingFrontend(sdh)
    
    with gr.Blocks() as demo: