↓ 97 callersMethodfrom_pretrained(cls, name, args=None, base_cls=None, *, home_path=None, url=None, prefix='', **kwargs)
anisoraV1_infer/sat/examples/chatglm/chat_model.py:43
↓ 64 callersMethodfrom_pretrained(cls, name, args=None, *, home_path=None, url=None, prefix='', build_only=False, use_node_group=True, overwrit
anisoraV1_infer/sat/model/base_model.py:214
↓ 42 callersMethodhead_to_batch_dimr""" Reshape the tensor from `[batch_size, seq_len, dim]` to `[batch_size, seq_len, heads, dim // heads]` `heads` is the number of hea
anisoraV1_infer/fastercache/models/vchitect/attention.py:624
↓ 39 callersMethod_transpose_for_scoresTranspose a 3D tensor [b, s, np*hn] into a 4D tensor with size [b, np, s, hn].
anisoraV1_train_npu/swissarmytransformer-npu_t_sp/sat/model/transformer.py:99
↓ 39 callersMethod_transpose_for_scoresTranspose a 3D tensor [b, s, np*hn] into a 4D tensor with size [b, np, s, hn].
anisoraV1_infer/sat/model/transformer.py:99
↓ 39 callersMethod_transpose_for_scoresTranspose a 3D tensor [b, s, np*hn] into a 4D tensor with size [b, np, s, hn].
anisoraV1_infer/sat/sat/model/transformer.py:99
↓ 39 callersMethod_transpose_for_scoresTranspose a 3D tensor [b, s, np*hn] into a 4D tensor with size [b, np, s, hn].
anisoraV1_train_gpu/swissarmytransformer-npu_t_sp/sat/model/transformer.py:99
↓ 36 callersMethod_transpose_for_scoresTranspose a 3D tensor [b, s, np*hn] into a 4D tensor with size [b, np, s, hn].
anisora_rl/SwissArmyTransformer-main/sat/model/transformer.py:99