mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-04-30 11:11:37 +00:00
More guidance work. Improved LoRA module resolver for unet. Added vega mappings and LoRA training for it. Various other bigfixes and changes
This commit is contained in:
@@ -46,8 +46,11 @@ sdxl_sampler_config = {
|
||||
|
||||
def get_sampler(
|
||||
sampler: str,
|
||||
kwargs: dict = None,
|
||||
):
|
||||
sched_init_args = {}
|
||||
if kwargs is not None:
|
||||
sched_init_args.update(kwargs)
|
||||
|
||||
if sampler.startswith("k_"):
|
||||
sched_init_args["use_karras_sigmas"] = True
|
||||
|
||||
Reference in New Issue
Block a user