bug fix + cleaned losses

code/losses.py

@@ -9,9 +9,6 @@ from shapely.geometry import Point
 from shapely.geometry.polygon import Polygon
 from torchvision import transforms
 from PIL import Image
-from transformers import CLIPProcessor, CLIPModel
-
-from diffusers import StableDiffusionPipeline
 
 class SDSLoss(nn.Module):
     def __init__(self, cfg, device, model):
@@ -20,11 +17,6 @@ class SDSLoss(nn.Module):
         self.device = device
         self.pipe = model
 
-        # self.clip_model = CLIPModel.from_pretrained("openai/clip-vit-large-patch14").to(self.device)
-        # self.clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-large-patch14")
-    
-        # default scheduler: PNDMScheduler(beta_start=0.00085, beta_end=0.012,
-        # beta_schedule="scaled_linear", num_train_timesteps=1000)
         self.alphas = self.pipe.scheduler.alphas_cumprod.to(self.device)
         self.sigmas = (1 - self.pipe.scheduler.alphas_cumprod).to(self.device)
 
@@ -46,9 +38,6 @@ class SDSLoss(nn.Module):
     
         self.text_embeddings = torch.cat([uncond_embeddings, text_embeddings])
         self.text_embeddings = self.text_embeddings.repeat_interleave(self.cfg.batch_size, 0)
-        del self.pipe.tokenizer
-        del self.pipe.text_encoder
-
 
     def forward(self, x_aug):
         sds_loss = 0