call sampler by name

This commit is contained in:
arcticfaded
2022-10-19 05:19:01 +00:00
parent e7f4808505
commit 0f0d6ab8e0
2 changed files with 9 additions and 8 deletions

View File

@@ -1,7 +1,7 @@
from inflection import underscore
from typing import Any, Dict, Optional
from pydantic import BaseModel, Field, create_model
from modules.processing import StableDiffusionProcessing, Processed, StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img, process_images
from modules.processing import StableDiffusionProcessingTxt2Img
import inspect
@@ -95,5 +95,5 @@ class PydanticModelGenerator:
StableDiffusionProcessingAPI = PydanticModelGenerator(
"StableDiffusionProcessingTxt2Img",
StableDiffusionProcessingTxt2Img,
[{"key": "sampler_index", "type": str, "default": "k_euler_a"}]
).generate_model()
[{"key": "sampler_index", "type": str, "default": "Euler"}]
).generate_model()