rich-text-to-image

Build error

bumsika commited on May 25, 2023

Commit

13c7cc2

1 Parent(s): 2dee308

tree.py

Files changed (1) hide show

models/region_diffusion.py CHANGED Viewed

@@ -17,7 +17,7 @@ class RegionDiffusion(nn.Module):
         super().__init__()
         self.device = device
-        self.num_train_timesteps = 1000
         self.clip_gradient = False
         print(f'[INFO] loading stable diffusion...')

         super().__init__()
         self.device = device
+        self.num_train_timesteps = 10
         self.clip_gradient = False
         print(f'[INFO] loading stable diffusion...')