callback error

[Neytiri7]

** Error executing callback cfg_denoised_callback for O:\AI\SynologyDrive\repo\extensions\sd_webui_SAG\scripts\SAG.py

Traceback (most recent call last):

  File "O:\AI\SynologyDrive\repo\modules\script_callbacks.py", line 203, in cfg_denoised_callback

    c.callback(params)

  File "O:\AI\SynologyDrive\repo\extensions\sd_webui_SAG\scripts\SAG.py", line 214, in denoised_callback

    attn_mask.reshape(b, middle_layer_latent_size[0], middle_layer_latent_size[1])

RuntimeError: shape '[1, 12, 8]' is invalid for input of size 384

---

*** Error executing callback cfg_after_cfg_callback for O:\AI\SynologyDrive\repo\extensions\sd_webui_SAG\scripts\SAG.py

Traceback (most recent call last):

  File "O:\AI\SynologyDrive\repo\modules\script_callbacks.py", line 211, in cfg_after_cfg_callback

    c.callback(params)

  File "O:\AI\SynologyDrive\repo\extensions\sd_webui_SAG\scripts\SAG.py", line 242, in cfg_after_cfg_callback

    params.x = params.x + (current_uncond_pred - (current_degraded_pred + current_degraded_pred_compensation)) * float(current_sag_guidance_scale)

RuntimeError: The size of tensor a (64) must match the size of tensor b (128) at non-singleton dimension 3