Spaces:

ironjr
/

SemanticPalette

Running on Zero

ironjr commited on Mar 19

Commit

13ba306

•

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!')