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/load_params.py
+1 -1
View File
@@ -129,7 +129,7 @@ def convert_and_save_torch_params(is_mega: bool, model_path: str):
encoder_params[i] = encoder_params[i].to(torch.float16)
detoker_params = load_vqgan_torch_params('./pretrained/vqgan')
detoker_path = os.path.join('pretrained', 'vqgan', 'detokenizer.pt')
detoker_path = os.path.join('pretrained', 'vqgan', 'detoker.pt')
torch.save(encoder_params, os.path.join(model_path, 'encoder.pt'))
torch.save(decoder_params, os.path.join(model_path, 'decoder.pt'))