Update min_dalle/min_dalle.py

min_dalle/min_dalle.py

@@ -14,8 +14,8 @@ import time
 
 torch.set_grad_enabled(False)
 torch.set_num_threads(os.cpu_count())
-torch.backends.cudnn.enabled = True
-torch.backends.cudnn.allow_tf32 = True
+torch.backends.cudnn.enabled = False
+torch.backends.cudnn.allow_tf32 = False
 
 MIN_DALLE_REPO = 'https://huggingface.co/kuprel/min-dalle/resolve/main/'
 IMAGE_TOKEN_COUNT = 256
@@ -252,8 +252,7 @@ class MinDalle:
                     token_index=token_indices[[i]]
                 )
                 
-                del attention_state
-                del image_tokens
+      
              
 
             with torch.cuda.amp.autocast(dtype=torch.float16):