Spaces:

fffiloni
/

imagic-stable-diffusion

Paused

fffiloni commited on Nov 30, 2022

Commit

d4c89e2

•

1 Parent(s): 8e52970

Update imagic.py

Files changed (1) hide show

imagic.py CHANGED Viewed

@@ -292,7 +292,7 @@ class ImagicStableDiffusionPipeline(DiffusionPipeline):
         # Now we fine tune the unet to better reconstruct the image
         self.unet.requires_grad_(True)
         self.unet.train()
-        optimizer = torch.optim.Adam(
             self.unet.parameters(),  # only optimize unet
             lr=diffusion_model_learning_rate,
         )

         # Now we fine tune the unet to better reconstruct the image
         self.unet.requires_grad_(True)
         self.unet.train()
+        optimizer = bnb.optim.Adam8bit(
             self.unet.parameters(),  # only optimize unet
             lr=diffusion_model_learning_rate,
         )