tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fixed wrong file path

Browse Source
This commit is contained in:
Brett Kuprel
2022-07-01 10:58:29 -04:00
parent fffd0f2b83
commit 7bf76deafb
3 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
min_dalle/min_dalle_torch.py
+1 -1
View File
@@ -27,7 +27,7 @@ class MinDalleTorch(MinDalleBase):
self.encoder_params_path = os.path.join(self.model_path, 'encoder.pt')
self.decoder_params_path = os.path.join(self.model_path, 'decoder.pt')
self.detoker_params_path = os.path.join('pretrained', 'vqgan', 'detokenizer.pt')
self.detoker_params_path = os.path.join('pretrained', 'vqgan', 'detoker.pt')
is_converted = os.path.exists(self.encoder_params_path)
is_converted &= os.path.exists(self.decoder_params_path)