Merge pull request #98 from donno2048/main

Trivial fixes

Author: lucidrains

deep_daze/clip.py

@@ -11,10 +11,7 @@
 import urllib
 import warnings
 from typing import Union, List
-
-import torch
-from PIL import Image
-from torchvision.transforms import Compose, Resize, CenterCrop, ToTensor, Normalize
+from torchvision.transforms import Compose, Normalize
 from tqdm import tqdm
 
 _MODELS = {
@@ -613,9 +610,7 @@ def build_model(state_dict: dict):
     model.load_state_dict(state_dict)
     return model.eval()
 
-import gzip
 import html
-import os
 from functools import lru_cache
 
 import ftfy
@@ -745,6 +740,4 @@ def decode(self, tokens):
         text = ''.join([self.decoder[token] for token in tokens])
         text = bytearray([self.byte_decoder[c] for c in text]).decode('utf-8', errors="replace").replace('</w>', ' ')
         return text
-        import gzip
-
 _tokenizer = SimpleTokenizer()

deep_daze/deep_daze.py

@@ -5,8 +5,6 @@
 import random
 from datetime import datetime
 from pathlib import Path
-from shutil import copy
-
 import torch
 import torch.nn.functional as F
 from siren_pytorch import SirenNet, SirenWrapper
@@ -16,7 +14,7 @@
 
 from PIL import Image
 import torchvision.transforms as T
-from torchvision.utils import save_image
+#from torchvision.utils import save_image
 
 from tqdm import trange, tqdm
 
@@ -74,7 +72,7 @@ def open_folder(path):
         cmd_list = ['xdg-open', path]
     elif sys.platform in ['win32', 'win64']:
         cmd_list = ['explorer', path.replace('/', '\\')]
-    if cmd_list == None:
+    if cmd_list is None:
         return
 
     try: