Update min_dalle/min_dalle.py

min_dalle/min_dalle.py

@@ -240,7 +240,7 @@ class MinDalle:
             st.session_state.bar.progress(i/IMAGE_TOKEN_COUNT)
 
             torch.cuda.empty_cache()   
-            torch.cpu.empty_cache()              
+            torch.device('cpu').empty_cache()            
             with torch.cuda.amp.autocast(dtype=self.dtype):
                 image_tokens[i + 1], attention_state = self.decoder.forward(
                     settings=settings,