Method__init__(
self, dim, ratio, max_position_embeddings=2048, base=10000, device=None
)
core_training/fastchat/model/llama_condense_monkey_patch.py:11
Method__init__(self, teacher, tokenizer, student_lm, K=128, beta=2.0, lambda_pref=2.0)
compression_training/compress.py:221
Method_forward_with_hidden_states(
self, input_ids, attention_mask, inputs_embeds, labels,
human_end_positions, prepend
eval/alfworld/eval_agent/agents/model_insert_hidden.py:1029
Functionadd_text(
state0, state1, model_selector0, model_selector1, text, request: gr.Request
)
core_training/fastchat/serve/gradio_block_arena_named.py:146
Functionadd_text(
state0, state1, model_selector0, model_selector1, text, request: gr.Request
)
core_training/fastchat/serve/gradio_block_arena_anony.py:483