fffiloni commited on
Commit
d4c89e2
1 Parent(s): 8e52970

Update imagic.py

Browse files
Files changed (1) hide show
  1. imagic.py +1 -1
imagic.py CHANGED
@@ -292,7 +292,7 @@ class ImagicStableDiffusionPipeline(DiffusionPipeline):
292
  # Now we fine tune the unet to better reconstruct the image
293
  self.unet.requires_grad_(True)
294
  self.unet.train()
295
- optimizer = torch.optim.Adam(
296
  self.unet.parameters(), # only optimize unet
297
  lr=diffusion_model_learning_rate,
298
  )
 
292
  # Now we fine tune the unet to better reconstruct the image
293
  self.unet.requires_grad_(True)
294
  self.unet.train()
295
+ optimizer = bnb.optim.Adam8bit(
296
  self.unet.parameters(), # only optimize unet
297
  lr=diffusion_model_learning_rate,
298
  )