↓ 3 callersMethodattn_batch(self, q, k, v, sim, attn, is_cross, place_in_unet, num_heads, **kwargs)
freecustom/mrsa.py:51
↓ 2 callersFunctionload_mask Load an image mask, resize it, and prepare it for use in PyTorch [0, 255] -> [0, 1], returned shape (1,1,H,W) Parameters: mask_path (
utils/utils.py:36