Update min_dalle/min_dalle.py

min_dalle/min_dalle.py

@@ -122,7 +122,6 @@ class MinDalle:
         del params
         self.encoder = self.encoder.to(device=self.device)
 
-    @st.cache
     def init_decoder(self):
         is_downloaded = os.path.exists(self.decoder_params_path)
         if not is_downloaded: self.download_decoder()
@@ -139,8 +138,7 @@ class MinDalle:
         self.decoder.load_state_dict(params, strict=False)
         del params
         self.decoder = self.decoder.to(device=self.device)
-
-    @st.cache
+        
     def init_detokenizer(self):
         is_downloaded = os.path.exists(self.detoker_params_path)
         if not is_downloaded: self.download_detokenizer()