Revert "Updated model location"

This reverts commit 4831980d34025aa262e0bd238cdf4dfcc0cc8e0c.

dalle/models/__init__.py

@@ -43,21 +43,20 @@ class Dalle(nn.Module):
     @classmethod
     def from_pretrained(cls,
                         path: str) -> nn.Module:
-        #path = _MODELS[path] if path in _MODELS else path
-        #path = utils.realpath_url_or_path(path, root=os.path.expanduser(".cache/minDALL-E"))
-        path = ".cache/minDALL-E/1.3B/"
+        path = _MODELS[path] if path in _MODELS else path
+        path = utils.realpath_url_or_path(path, root=os.path.expanduser(".cache/minDALL-E"))
 
         config_base = get_base_config()
-        config_new = OmegaConf.load(path+'config.yaml')
+        config_new = OmegaConf.load(os.path.join(path, 'config.yaml'))
         config_update = OmegaConf.merge(config_base, config_new)
 
         model = cls(config_update)
-        model.tokenizer = build_tokenizer(path+'tokenizer',
+        model.tokenizer = build_tokenizer(os.path.join(path, 'tokenizer'),
                                           context_length=model.config_dataset.context_length,
                                           lowercase=True,
                                           dropout=None)
-        model.stage1.from_ckpt(path+'stage1_last.ckpt')
-        model.stage2.from_ckpt(path+'stage2_last.ckpt')
+        model.stage1.from_ckpt(os.path.join(path, 'stage1_last.ckpt'))
+        model.stage2.from_ckpt(os.path.join(path, 'stage2_last.ckpt'))
         return model
 
     @torch.no_grad()