Skip to content

Commit

Permalink
add autoedit script
Browse files Browse the repository at this point in the history
  • Loading branch information
bmtm committed Apr 17, 2022
1 parent 7d059d6 commit a0b5be4
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -75,6 +75,10 @@ python sample.py --mask mask.png --model_path inpaint.pt --edit output_npy/00000
# additional arguments for uncropping
python sample.py --edit_x 64 --edit_y 64 --edit_width 128 --edit_height 128 --model_path inpaint.pt --edit output_npy/00000.npy --batch_size 6 --num_batches 6 --text "your prompt"
# autoedit uses the inpaint model to give the ldm an image prompting function (that works differently from --init_image)
# it continuously edits random parts of the image to maximize clip score for the text prompt
python autoedit.py --edit image.png --model_path inpaint.pt --batch_size 6 --text "your prompt"
```

# Training/Fine tuning
Expand Down

0 comments on commit a0b5be4

Please sign in to comment.