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

it was correct the first time, my bad

Browse Source
This commit is contained in:
Phil Wang
2022-05-20 18:05:15 -07:00
parent 721f9687c1
commit 430961cb97
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
dalle2_pytorch/trackers.py
View File

@@ -1,6 +1,5 @@
import os import os
from pathlib import Path from pathlib import Path
from enum import Enum
import importlib import importlib
from itertools import zip_longest from itertools import zip_longest
@@ -105,7 +104,7 @@ class WandbTracker(BaseTracker):
Takes a tensor of images and a list of captions and logs them to wandb. Takes a tensor of images and a list of captions and logs them to wandb.
""" """
wandb_images = [self.wandb.Image(image, caption=caption) for image, caption in zip_longest(images, captions)] wandb_images = [self.wandb.Image(image, caption=caption) for image, caption in zip_longest(images, captions)]
self.wandb.log({ image_section: wandb_images }, **kwargs) self.log({ image_section: wandb_images }, **kwargs)
def save_state_dict(self, state_dict, relative_path, **kwargs): def save_state_dict(self, state_dict, relative_path, **kwargs):
""" """

2
setup.py
View File

@@ -10,7 +10,7 @@ setup(
'dream = dalle2_pytorch.cli:dream' 'dream = dalle2_pytorch.cli:dream'
], ],
}, },
version = '0.3.4', version = '0.3.5',
license='MIT', license='MIT',
description = 'DALL-E 2', description = 'DALL-E 2',
author = 'Phil Wang', author = 'Phil Wang',