updated

Procfile

@@ -1 +1 @@
-web: sh setup.sh && streamlit run streamlit_app.py
+web: streamlit run streamlit_app.py

dalle/models/__init__.py

@@ -57,7 +57,7 @@ class Dalle(nn.Module):
                                           lowercase=True,
                                           dropout=None)
         model.stage1.from_ckpt('.cache/minDALL-E/1.3B/stage1_last.ckpt')
-        model.stage2.from_ckpt('https://utexas.box.com/shared/static/54jc9fw0bious5nx6wvayeqaskcrdgv4.ckpt')
+        model.stage2.from_ckpt('.cache/minDALL-E/1.3B/stage2_last.ckpt')
         #model.stage1.from_ckpt('https://utexas.box.com/shared/static/rpt9miyj2kikogyekpqnkd6y115xp51i.ckpt')
         #model.stage2.from_ckpt('https://utexas.box.com/shared/static/54jc9fw0bious5nx6wvayeqaskcrdgv4.ckpt')
         

dalle/models/stage1/vqgan.py

@@ -92,8 +92,8 @@ class VQGAN(nn.Module):
         #self.load_state_dict(ckpt, strict=strict)
         #print(f'{path} successfully restored..')
 
-        #ckpt = torch.load(path, map_location='cpu')['state_dict']
-        ckpt = torch.utils.model_zoo.load_url('https://utexas.box.com/shared/static/rpt9miyj2kikogyekpqnkd6y115xp51i.ckpt', map_location='cpu')['state_dict']
+        ckpt = torch.load(path, map_location='cpu')['state_dict']
+        #ckpt = torch.utils.model_zoo.load_url('https://utexas.box.com/shared/static/rpt9miyj2kikogyekpqnkd6y115xp51i.ckpt', map_location='cpu')['state_dict']
         
         self.load_state_dict(ckpt, strict=True)
         print(f'{path} succesfully restored..')

dalle/models/stage2/transformer.py

@@ -143,9 +143,8 @@ class Transformer1d(nn.Module):
             return logits, presents
 
     def from_ckpt(self, path: str) -> None:
-        #ckpt = torch.load(path, map_location='cpu')['state_dict']
-        ckpt = torch.utils.model_zoo.load_url('https://utexas.box.com/shared/static/54jc9fw0bious5nx6wvayeqaskcrdgv4.ckpt', map_location='cpu')['state_dict']
-        
+        ckpt = torch.load(path, map_location='cpu')['state_dict']
+        #ckpt = torch.utils.model_zoo.load_url('https://utexas.box.com/shared/static/54jc9fw0bious5nx6wvayeqaskcrdgv4.ckpt', map_location='cpu')['state_dict']
         self.load_state_dict(ckpt, strict=True)
         print(f'{path} succesfully restored..')
 

requirements.txt

@@ -1,18 +1,10 @@
-clip==0.2.0
-Cython==0.29.30
-clip_anytorch==2.4.0
+streamlit==1.10.0
+torch==1.11.0
 htbuilder==0.6.0
-iteration_utilities==0.11.0
-numpy==1.22.4
+torchvision==0.12.0
+ftfy==6.0.3
+regex==2022.1.18
+pytorch_lightning==1.5.9
 omegaconf==2.2.2
-pages==0.3
-pandas==1.4.2
-Pillow==9.2.0
-pytorch_lightning==1.6.3
-ruclip==0.0.1
-rudalle==1.1.3
-streamlit==1.10.0
+einops==0.4.1
 tokenizers==0.12.1
-torch==1.8.0
-torchvision==0.9.0
-tqdm==4.64.0