Updated model location

dalle/models/__init__.py

@@ -43,20 +43,21 @@ 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 = _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/"
 
         config_base = get_base_config()
-        config_new = OmegaConf.load(os.path.join(path, 'config.yaml'))
+        config_new = OmegaConf.load(path+'config.yaml')
         config_update = OmegaConf.merge(config_base, config_new)
 
         model = cls(config_update)
-        model.tokenizer = build_tokenizer(os.path.join(path, 'tokenizer'),
+        model.tokenizer = build_tokenizer(path+'tokenizer',
                                           context_length=model.config_dataset.context_length,
                                           lowercase=True,
                                           dropout=None)
-        model.stage1.from_ckpt(os.path.join(path, 'stage1_last.ckpt'))
-        model.stage2.from_ckpt(os.path.join(path, 'stage2_last.ckpt'))
+        model.stage1.from_ckpt(path+'stage1_last.ckpt')
+        model.stage2.from_ckpt(path+'stage2_last.ckpt')
         return model
 
     @torch.no_grad()