mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-10-31 03:20:40 +00:00
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
from imaginairy.imagine import imagine_images, imagine_image_files
|
|
from imaginairy.schema import ImaginePrompt, WeightedPrompt
|
|
from . import TESTS_FOLDER
|
|
|
|
|
|
def test_imagine():
|
|
prompt = ImaginePrompt(
|
|
"a scenic landscape", width=512, height=256, steps=20, seed=1
|
|
)
|
|
result = next(imagine_images(prompt))
|
|
assert result.md5() == "4c5957c498881d365cfcf13014812af0"
|
|
result.img.save(f"{TESTS_FOLDER}/test_output/scenic_landscape.png")
|
|
|
|
|
|
def test_img_to_img():
|
|
prompt = ImaginePrompt(
|
|
"a photo of a beach",
|
|
init_image=f"{TESTS_FOLDER}/data/beach_at_sainte_adresse.jpg",
|
|
init_image_strength=0.5,
|
|
width=512,
|
|
height=512,
|
|
steps=50,
|
|
seed=1,
|
|
sampler_type="DDIM",
|
|
)
|
|
out_folder = f"{TESTS_FOLDER}/test_output"
|
|
out_folder = "/home/bryce/Mounts/drennanfiles/art/tests"
|
|
imagine_image_files(prompt, outdir=out_folder)
|
|
|
|
|
|
def test_img_to_file():
|
|
prompt = ImaginePrompt(
|
|
[
|
|
WeightedPrompt(
|
|
"an old growth forest, diffuse light poking through the canopy. high-resolution, nature photography, nat geo photo"
|
|
)
|
|
],
|
|
# init_image=f"{TESTS_FOLDER}/data/beach_at_sainte_adresse.jpg",
|
|
init_image_strength=0.5,
|
|
width=512 + 64,
|
|
height=512 - 64,
|
|
steps=50,
|
|
# seed=2,
|
|
sampler_type="PLMS",
|
|
upscale=True,
|
|
)
|
|
out_folder = f"{TESTS_FOLDER}/test_output"
|
|
out_folder = "/home/bryce/Mounts/drennanfiles/art/tests"
|
|
imagine_image_files(prompt, outdir=out_folder)
|
|
|
|
|
|
def test_img_conditioning():
|
|
prompt = ImaginePrompt(
|
|
"photo",
|
|
init_image=f"{TESTS_FOLDER}/data/beach_at_sainte_adresse.jpg",
|
|
init_image_strength=0.5,
|
|
width=512 + 64,
|
|
height=512 - 64,
|
|
steps=50,
|
|
# seed=2,
|
|
sampler_type="PLMS",
|
|
upscale=True,
|
|
)
|
|
out_folder = f"{TESTS_FOLDER}/test_output"
|
|
out_folder = "/home/bryce/Mounts/drennanfiles/art/tests"
|
|
imagine_image_files(prompt, outdir=out_folder, record_steps=True)
|