(
self,
model,
positive,
negative,
skip_steps_sigma_threshold,
cfg_star_rescale,
sigmas,
cfg_values,
stg_scale_values,
stg_rescale_values,
stg_layers_indices,
preset=None,
apply_apg=False,
apg_cfg_scale=1.0,
eta=1.0,
norm_threshold=0.0,
)
| 703 | return result |
| 704 | |
| 705 | def get_guider( |
| 706 | self, |
| 707 | model, |
| 708 | positive, |
| 709 | negative, |
| 710 | skip_steps_sigma_threshold, |
| 711 | cfg_star_rescale, |
| 712 | sigmas, |
| 713 | cfg_values, |
| 714 | stg_scale_values, |
| 715 | stg_rescale_values, |
| 716 | stg_layers_indices, |
| 717 | preset=None, |
| 718 | apply_apg=False, |
| 719 | apg_cfg_scale=1.0, |
| 720 | eta=1.0, |
| 721 | norm_threshold=0.0, |
| 722 | ): |
| 723 | if preset and preset != "Custom": |
| 724 | preset_data = next( |
| 725 | (item for item in STG_ADVANCED_PRESETS if item["name"] == preset), None |
| 726 | ) |
| 727 | if preset_data: |
| 728 | skip_steps_sigma_threshold = preset_data["skip_steps_sigma_threshold"] |
| 729 | cfg_star_rescale = preset_data["cfg_star_rescale"] |
| 730 | sigma_list = preset_data["sigmas"] |
| 731 | cfg_list = preset_data["cfg_values"] |
| 732 | stg_scale_list = preset_data["stg_scale_values"] |
| 733 | stg_rescale_list = preset_data["stg_rescale_values"] |
| 734 | stg_layers_indices_list = preset_data["stg_layers_indices"] |
| 735 | else: |
| 736 | raise ValueError(f"Preset {preset} not found in the presets list.") |
| 737 | |
| 738 | else: |
| 739 | sigma_list = [float(s.strip()) for s in sigmas.split(",")] |
| 740 | cfg_list = [float(c.strip()) for c in cfg_values.split(",")] |
| 741 | stg_scale_list = [float(s.strip()) for s in stg_scale_values.split(",")] |
| 742 | stg_rescale_list = [float(s.strip()) for s in stg_rescale_values.split(",")] |
| 743 | stg_layers_indices_list = self.parse_stg_layers_indices(stg_layers_indices) |
| 744 | print("Using preset: ", preset) |
| 745 | print("Skip steps sigma threshold: ", skip_steps_sigma_threshold) |
| 746 | print("Cfg star rescale: ", cfg_star_rescale) |
| 747 | print("Sigma list: ", sigma_list) |
| 748 | print("Cfg list: ", cfg_list) |
| 749 | print("Stg scale list: ", stg_scale_list) |
| 750 | print("Stg rescale list: ", stg_rescale_list) |
| 751 | print("Stg layers indices list: ", stg_layers_indices_list) |
| 752 | |
| 753 | guider = STGGuiderAdvanced( |
| 754 | model, |
| 755 | sigma_list, |
| 756 | cfg_list, |
| 757 | stg_scale_list, |
| 758 | stg_rescale_list, |
| 759 | stg_layers_indices_list, |
| 760 | skip_steps_sigma_threshold, |
| 761 | cfg_star_rescale, |
| 762 | apply_apg, |
nothing calls this directly
no test coverage detected