| 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", | |
| ] | |