(self, x)

( self, x, attn_bias, using_nattn=False, kernel_size=7, )

( self, x, cond_BD, attn_bias, using_nattn=False, kernel_size=

(self, x_BLC: torch.Tensor, cond_BD: torch.Tensor)

(self, inp, ret_usages=False)

()

()

()

(self)

( self, ms_idx_Bl: List[torch.Tensor], same_shape: bool, last_one=False )

( self, x, v_patch_nums: Optional[Sequence[Union[int, Tuple[int, int]]]] = None,

()

(local_out_path, only_sync_master=False, timeout=30)

(fork=False, backend='nccl', gpu_id_if_not_distibuted=0, timeout=30)

(self, d: Union[OrderedDict, dict, str])

(self, state, strict=True)

(func)

(self, tag, img_chw, step=None)

(self, tag, tensor1d, step=None)

(func)

(self)

(ranks: List[int])

(x)

(x)

(cmd)

(cmd)

(path)

(*args, **kwargs)

(self, batch_size: int)

(self, batch_size: int)

(self)

(self)

(*args, **kwargs)

(gpu_id: int)

(self, key_ordered=True)

(self)

(self, head="scalar", step=None, **kwargs)

(self)

(self)

(*args, **kwargs)