Spaces:

ironjr
/

semantic-draw-canvas-sd15

Runtime error

ironjr commited on Mar 19, 2024

Commit

13ba306

verified ·

1 Parent(s): ceffafa

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -166,7 +166,7 @@ class StableMultiDiffusionPipeline(nn.Module):
         self.vae_scale_factor = self.pipe.vae_scale_factor
         # Prepare white background for bootstrapping.
-        self.get_white_background(768, 768)
         print(f'[INFO] Model is loaded!')

         self.vae_scale_factor = self.pipe.vae_scale_factor
         # Prepare white background for bootstrapping.
+        # self.get_white_background(768, 768) # This cause problem in HF ZeroGPU environment.
         print(f'[INFO] Model is loaded!')