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

update readme

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-08 06:01:07 -04:00
parent 78c20ce731
commit e409c120d0
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
README.md vendored
View File

@ -68,15 +68,14 @@ images = model.generate_images(
)
```
Note: you will have to move the images to the cpu and convert to numpy, e.g.
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()
```
Then image $i$ can be coverted to a PIL.Image
Then image $i$ can be coverted to a PIL.Image and saved with its `save` method
```python
image = Image.fromarray(images[i])
```
and saved with its `save` method `image.save('image.png')`.
### Interactive