aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-19 17:54:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix path_or_state bug

Browse Source
This commit is contained in:
Phil Wang
2022-07-26 09:47:54 -07:00
parent 62043acb2f
commit 9646dfc0e6
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dalle2_pytorch/trainer.py
View File

@@ -300,7 +300,7 @@ class DiffusionPriorTrainer(nn.Module):
# all processes need to load checkpoint. no restriction here # all processes need to load checkpoint. no restriction here
if isinstance(path_or_state, str): if isinstance(path_or_state, str):
path = Path(path) path = Path(path_or_state)
assert path.exists() assert path.exists()
loaded_obj = torch.load(str(path), map_location=self.device) loaded_obj = torch.load(str(path), map_location=self.device)

2
dalle2_pytorch/version.py
View File

@@ -1 +1 @@
__version__ = '1.0.5' __version__ = '1.0.6'