Browse Source

add border example

pull/2/head
oriBetelgeuse 5 years ago
parent
commit
9812794e47
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      rudalle/image_prompts.py

+ 1
- 0
rudalle/image_prompts.py View File

@ -9,6 +9,7 @@ class ImagePrompts:
Args:
pil_image (PIL.Image): image in PIL format
borders (dict[str] | int): borders that we croped from pil_image
example: {'up': 4, 'right': 0, 'left': 0, 'down': 0} (1 int eq 8 pixels)
vae (VQGanGumbelVAE): VQGAN model for image encoding
device (str): cpu or cuda
crop_first (bool): if True, croped image before VQGAN encoding


Loading…
Cancel
Save