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

detach unnecessary

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-08 13:26:53 -04:00
parent 1b45a5de2a
commit 703bfb231d
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md vendored
View File

@ -70,9 +70,9 @@ images = model.generate_images(
To get an image into PIL format you will have to first move the images to the CPU and convert the tensor to a numpy array.
```python
images = images.to('cpu').detach().numpy()
images = images.to('cpu').numpy()
```
Then image $i$ can be coverted to a PIL.Image and saved with its `save` method
Then image $i$ can be coverted to a PIL.Image and saved
```python
image = Image.fromarray(images[i])
image.save('image_{}.png'.format(i))

2
min_dalle/min_dalle.py
View File

@ -161,7 +161,7 @@ class MinDalle:
grid_size = int(sqrt(images.shape[0]))
images = images.reshape([grid_size] * 2 + list(images.shape[1:]))
image = images.flatten(1, 2).transpose(0, 1).flatten(1, 2)
image = Image.fromarray(image.to('cpu').detach().numpy())
image = Image.fromarray(image.to('cpu').numpy())
return image