require

sgm/models/diffusion.py

@@ -5,7 +5,6 @@ import pytorch_lightning as pl
 import torch
 from omegaconf import ListConfig, OmegaConf
 from safetensors.torch import load_file as load_safetensors
-from torch.optim.lr_scheduler import LambdaLR
 
 from ..modules import UNCONDITIONAL_CONFIG
 from ..modules.diffusionmodules.wrappers import OPENAIUNETWRAPPER

sgm/modules/diffusionmodules/loss.py

@@ -4,7 +4,7 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 from omegaconf import ListConfig
-from taming.modules.losses.lpips import LPIPS
+# from taming.modules.losses.lpips import LPIPS
 from torchvision.utils import save_image
 from ...util import append_dims, instantiate_from_config
 
@@ -26,8 +26,8 @@ class StandardDiffusionLoss(nn.Module):
         self.type = type
         self.offset_noise_level = offset_noise_level
 
-        if type == "lpips":
-            self.lpips = LPIPS().eval()
+        # if type == "lpips":
+        #     self.lpips = LPIPS().eval()
 
         if not batch2model_keys:
             batch2model_keys = []

sgm/modules/diffusionmodules/openaimodel.py

@@ -1,4 +1,3 @@
-import os
 import math
 from abc import abstractmethod
 from functools import partial

sgm/modules/diffusionmodules/sampling.py

@@ -7,7 +7,6 @@ from typing import Dict, Union
 
 import imageio
 import torch
-import json
 import numpy as np
 import torch.nn.functional as F
 from omegaconf import ListConfig, OmegaConf