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 05:59:06 -04:00
parent 870cebdaaf
commit 78c20ce731
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
README.md vendored
View File

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