File size: 2,346 Bytes
1e3b872 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
import json
import os
from prompt_parser import parse_workflow
current_path = os.path.dirname(os.path.abspath(__file__))
def test_prompt_parser_02():
path = f"{current_path}/examples/02.json"
prompt_info = open(path, "rt").read()
parsed = parse_workflow(json.loads(prompt_info))
assert parsed == [
"CLIPTextEncode->clip->LoraLoader->lora_name::oc_v10.safetensors",
"CLIPTextEncode->clip->LoraLoader->strength_clip::0.5",
"CLIPTextEncode->clip->LoraLoader->strength_model::0.5",
"ImageUpscaleWithModel->upscale_model->UpscaleModelLoader->model_name::RealESRGAN_x4plus_anime_6B.pth",
"KSampler->latent_image->EmptyLatentImage->batch_size::1",
"KSampler->latent_image->EmptyLatentImage->height::512",
"KSampler->latent_image->EmptyLatentImage->width::768",
"KSampler->model->LoraLoader->lora_name::oc_v10.safetensors",
"KSampler->model->LoraLoader->strength_clip::0.5",
"KSampler->model->LoraLoader->strength_model::0.5",
"KSampler->negative->CLIPTextEncode->text::embeddings:EasyNegative, (worst quality, low quality, normal quality, bad anatomy:1.4), text, watermark,",
"KSampler->positive->CLIPTextEncode->text::1girl, solo, expressionless, silver hair, ponytail, purple eyes, face mask, ninja, blue bikini, medium breats, cleavage, scarf,",
"LoraLoader->clip->CheckpointLoaderSimple->ckpt_name::agelesnate_v121.safetensors",
"LoraLoader->model->CheckpointLoaderSimple->ckpt_name::agelesnate_v121.safetensors",
"VAEDecode->samples->KSampler->cfg::1.5",
"VAEDecode->samples->KSampler->cfg::7",
"VAEDecode->samples->KSampler->denoise::0.5",
"VAEDecode->samples->KSampler->denoise::1",
"VAEDecode->samples->KSampler->sampler_name::dpmpp_2m_sde",
"VAEDecode->samples->KSampler->scheduler::karras",
"VAEDecode->samples->KSampler->seed::804645408561758",
"VAEDecode->samples->KSampler->steps::20",
"VAEDecode->samples->KSampler->steps::40",
"VAEDecode->vae->VAELoader->vae_name::vae-ft-mse-840000-ema-pruned.safetensors",
"VAEEncode->pixels->ImageScaleBy->scale_by::0.5",
"VAEEncode->pixels->ImageScaleBy->upscale_method::area",
"VAEEncode->vae->VAELoader->vae_name::vae-ft-mse-840000-ema-pruned.safetensors",
]
|