Update imagic.py

imagic.py

@@ -5,7 +5,7 @@
 import inspect
 import warnings
 from typing import List, Optional, Union
-
+import bitsandbytes as bnb
 import numpy as np
 import torch
 import torch.nn.functional as F
@@ -236,7 +236,8 @@ class ImagicStableDiffusionPipeline(DiffusionPipeline):
         text_embeddings_orig = text_embeddings.clone()
 
         # Initialize the optimizer
-        optimizer = torch.optim.Adam(
+        
+        optimizer = bnb.optim.Adam8bit(
             [text_embeddings],  # only optimize the embeddings
             lr=embedding_learning_rate,
         )