SYMBOL INDEX (637 symbols across 98 files)

FILE: examples/fedmkt/fedmkt.py
  function main (line 12) | def main(config="./config.yaml", param: Union[Dict, str] = None):

FILE: examples/offsite_tuning/offsite_tuning.py
  function load_params (line 11) | def load_params(file_path):
  function setup_pipeline (line 17) | def setup_pipeline(params):
  function main (line 112) | def main(config_file, param_file):

FILE: examples/pellm/test_bloom_lora.py
  function main (line 14) | def main(config="../../config.yaml", param: Union[Dict, str] = None, nam...

FILE: python/fate_llm/algo/dp/dp_trainer.py
  class DPTrainingArguments (line 32) | class DPTrainingArguments(Seq2SeqTrainingArguments):
  class DPTrainer (line 39) | class DPTrainer(object):
    method __init__ (line 40) | def __init__(
    method _init_dp_model (line 78) | def _init_dp_model(self):
    method train (line 97) | def train(self):
    method _train_an_epoch (line 103) | def _train_an_epoch(self):
    method _prepare_batch_input (line 141) | def _prepare_batch_input(self, input_ids) -> dict:
    method freeze_model_embedding (line 148) | def freeze_model_embedding(self):
    method save_model (line 151) | def save_model(

FILE: python/fate_llm/algo/dp/opacus_compatibility/__init__.py
  function add_layer_compatibility (line 20) | def add_layer_compatibility(opacus):
  function add_optimizer_compatibility (line 30) | def add_optimizer_compatibility(optimizer):

FILE: python/fate_llm/algo/dp/opacus_compatibility/grad_sample/embedding.py
  function compute_embedding_grad_sample (line 24) | def compute_embedding_grad_sample(

FILE: python/fate_llm/algo/dp/opacus_compatibility/optimizers/optimizer.py
  function add_noise (line 27) | def add_noise(self):
  function add_noise_wrapper (line 47) | def add_noise_wrapper(optimizer):

FILE: python/fate_llm/algo/dp/opacus_compatibility/transformers_compate.py
  function get_model_class (line 22) | def get_model_class(model):
  function prepare_position_ids (line 31) | def prepare_position_ids(model, input_ids):
  function _get_position_ids_for_gpt2 (line 38) | def _get_position_ids_for_gpt2(input_ids):

FILE: python/fate_llm/algo/fdkt/cluster/cluster.py
  class SentenceCluster (line 20) | class SentenceCluster(object):
    method __init__ (line 21) | def __init__(self, model, cluster_method="kmeans", n_clusters=8, **oth...
    method get_embeddings (line 27) | def get_embeddings(self, sentences: List[str]):
    method cluster (line 30) | def cluster(self, sentences):

FILE: python/fate_llm/algo/fdkt/cluster/cluster_method.py
  class KMeansRunner (line 19) | class KMeansRunner(object):
    method __init__ (line 20) | def __init__(self, n_clusters, **other_cluster_args):
    method fit (line 24) | def fit(self, x):
  function get_cluster_runner (line 31) | def get_cluster_runner(method, n_clusters, **other_cluster_args):

FILE: python/fate_llm/algo/fdkt/fdkt_data_aug.py
  class FDKTTrainingArguments (line 38) | class FDKTTrainingArguments(Seq2SeqTrainingArguments):
    method to_dict (line 76) | def to_dict(self):
  class FDKTSLM (line 91) | class FDKTSLM(object):
    method __init__ (line 92) | def __init__(
    method aug_data (line 117) | def aug_data(self):
    method dp_train (line 151) | def dp_train(self):
    method _create_inference_inst (line 182) | def _create_inference_inst(self):
    method _destroy_inference_inst (line 199) | def _destroy_inference_inst(self):
    method sync_synthetic_dataset (line 205) | def sync_synthetic_dataset(self, data):
    method sync_aug_data (line 208) | def sync_aug_data(self):
    method save_model (line 211) | def save_model(
  class FDKTLLM (line 223) | class FDKTLLM(object):
    method __init__ (line 224) | def __init__(
    method sync_synthetic_data (line 248) | def sync_synthetic_data(self):
    method sync_aug_data (line 251) | def sync_aug_data(self, aug_data):
    method aug_data (line 254) | def aug_data(self):
    method _aug (line 273) | def _aug(self, aug_prompts):
    method filter_data (line 289) | def filter_data(self, slm_data):
    method cluster_data (line 314) | def cluster_data(self, slm_data):

FILE: python/fate_llm/algo/fdkt/inference_inst.py
  function api_init (line 16) | def api_init(api_url: str, model_name: str, api_key: str = 'EMPTY', api_...
  function vllm_init (line 26) | def vllm_init(model_path: str, num_gpu=1, dtype='float16', gpu_memory_ut...

FILE: python/fate_llm/algo/fdkt/utils/dp_loss.py
  class SequenceCrossEntropyLoss (line 25) | class SequenceCrossEntropyLoss(nn.Module):
    method __init__ (line 26) | def __init__(self, model_type, label_smoothing=-1, reduce=None):
    method forward (line 32) | def forward(self, logits, targets, mask):
  function sequence_cross_entropy_with_logits (line 36) | def sequence_cross_entropy_with_logits(logits, targets, mask, label_smoo...

FILE: python/fate_llm/algo/fdkt/utils/invalid_data_filter.py
  function filter_invalid_data (line 20) | def filter_invalid_data(data_dict):

FILE: python/fate_llm/algo/fdkt/utils/text_generate.py
  function slm_text_generate (line 20) | def slm_text_generate(
  function general_text_generate (line 70) | def general_text_generate(

FILE: python/fate_llm/algo/fedavg/fedavg.py
  class Seq2SeqFedAVGClient (line 39) | class Seq2SeqFedAVGClient(HomoSeq2SeqTrainerClient):
    method __init__ (line 41) | def __init__(
    method init_aggregator (line 83) | def init_aggregator(self, ctx: Context, fed_args: FedArguments):
    method on_federation (line 91) | def on_federation(

FILE: python/fate_llm/algo/fedcollm/fedcollm.py
  class FedCoLLMBase (line 38) | class FedCoLLMBase(object):
    method update_model (line 40) | def update_model(model, updated_params):
  class SLM (line 45) | class SLM(FedCoLLMBase):
    method __init__ (line 46) | def __init__(
    method train (line 85) | def train(self):
    method _sync_slm_updated_params (line 111) | def _sync_slm_updated_params(self, iter_ctx):
    method _get_slm_training_args (line 115) | def _get_slm_training_args(self):
    method _init_aggregator (line 118) | def _init_aggregator(self, ctx: Context, fed_args: FedArguments):
  class LLM (line 128) | class LLM(FedCoLLMBase):
    method __init__ (line 129) | def __init__(
    method _init_aggregator (line 178) | def _init_aggregator(self, ctx: Context):
    method _get_logits (line 181) | def _get_logits(self, model):
    method on_epoch_begin (line 198) | def on_epoch_begin(self, iter_ctx):
    method _sync_slm_updated_params (line 203) | def _sync_slm_updated_params(self, iter_ctx):
    method _train_slm (line 209) | def _train_slm(self, iter_ctx, llm_pub_logits, epoch_idx):
    method _train_llm (line 242) | def _train_llm(self, slm_pub_logits, epoch_idx):
    method train (line 273) | def train(self):

FILE: python/fate_llm/algo/fedcollm/fedcollm_trainer.py
  function computing_kd_loss (line 34) | def computing_kd_loss(src_logits, dst_logits, loss_mask):
  function recovery_logits (line 45) | def recovery_logits(
  class FedCoLLMTrainer (line 73) | class FedCoLLMTrainer(Seq2SeqTrainer):
    method __init__ (line 82) | def __init__(self, **kwargs):
    method compute_loss (line 98) | def compute_loss(self,  model, inputs, return_outputs=False):

FILE: python/fate_llm/algo/fedcollm/fedcollm_training_args.py
  class FedCoLLMTrainingArguments (line 21) | class FedCoLLMTrainingArguments(Seq2SeqTrainingArguments):
    method to_dict (line 44) | def to_dict(self):
    method _pop_extra (line 58) | def _pop_extra(self):
    method to_slm_seq_training_args (line 65) | def to_slm_seq_training_args(self):
    method to_fedco_slm_training_args (line 71) | def to_fedco_slm_training_args(self):
    method to_fedco_llm_training_args (line 77) | def to_fedco_llm_training_args(self):

FILE: python/fate_llm/algo/fedcot/encoder_decoder/init/default_init.py
  class FedCoTEDAPIClientInit (line 22) | class FedCoTEDAPIClientInit(InferInit):
    method __init__ (line 28) | def __init__(self, ctx):
    method get_inst (line 32) | def get_inst(self):
  class FedCoTEDAPIServerInit (line 38) | class FedCoTEDAPIServerInit(InferInit):
    method __init__ (line 44) | def __init__(self, ctx):
    method get_inst (line 48) | def get_inst(self):

FILE: python/fate_llm/algo/fedcot/encoder_decoder/slm_encoder_decoder.py
  class SLMEncoderDecoderClient (line 33) | class SLMEncoderDecoderClient(InferDPTClient):
    method __init__ (line 35) | def __init__(self, ctx: Context, local_inference_inst: Inference) -> N...
    method encode (line 41) | def encode(self, docs: List[Dict[str, str]], format_template: str = No...
    method decode (line 57) | def decode(self, p_docs: List[Dict[str, str]], instruction_template: s...
    method inference (line 62) | def inference(self, docs: Union[List[Dict[str, str]], HuggingfaceDatas...
  class SLMEncoderDecoderServer (line 78) | class SLMEncoderDecoderServer(InferDPTServer):

FILE: python/fate_llm/algo/fedcot/fedcot_trainer.py
  function save_to (line 45) | def save_to(obj, filepath, filename='tmp.pkl'):
  function load (line 55) | def load(filepath, filename='tmp.pkl'):
  class DSSTrainerClient (line 71) | class DSSTrainerClient(Seq2SeqTrainer):
    method __init__ (line 73) | def __init__(self,
    method compute_loss (line 103) | def compute_loss(self, model, inputs, return_outputs=False):
  class FedCoTTrainerClient (line 111) | class FedCoTTrainerClient(DSSTrainerClient):
    method __init__ (line 113) | def __init__(self,
    method infer (line 182) | def infer(self) -> List[str]:
    method train (line 207) | def train(self):
    method get_infer_result (line 220) | def get_infer_result(self):
  class FedCoTTraineServer (line 224) | class FedCoTTraineServer(object):
    method __init__ (line 226) | def __init__(self, ctx: Context, infer_server: Union[SLMEncoderDecoder...
    method train (line 231) | def train(self):

FILE: python/fate_llm/algo/fedcot/slm_encoder_decoder_trainer.py
  class EDPrefixDataCollator (line 7) | class EDPrefixDataCollator(DataCollatorForSeq2Seq):
    method __call__ (line 8) | def __call__(self, features, return_tensors=None):
  class EncoderDecoderPrefixTrainer (line 19) | class EncoderDecoderPrefixTrainer(Seq2SeqTrainer):
    method __init__ (line 21) | def __init__(self, alpha=0.5, *args, **kwargs):
    method compute_loss (line 25) | def compute_loss(self, model, inputs, return_outputs=False):

FILE: python/fate_llm/algo/fedkseed/args.py
  class KSeedTrainingArguments (line 5) | class KSeedTrainingArguments:

FILE: python/fate_llm/algo/fedkseed/fedkseed.py
  class Trainer (line 17) | class Trainer:
    method __init__ (line 18) | def __init__(
    method get_clients (line 30) | def get_clients(ctx: Context):
    method load_model (line 38) | def load_model(self):
    method train (line 41) | def train(self):
    method should_stop (line 87) | def should_stop(self):
    method evaluate (line 90) | def evaluate(self):
  class ClientTrainer (line 94) | class ClientTrainer:
    method __init__ (line 95) | def __init__(self, ctx: Context, model, fedkseed_args, training_args, ...
    method train (line 108) | def train(self):
    method train_once (line 130) | def train_once(self, seed_candidates, seed_probabilities, direction_de...
  class FedKSeedTrainingArguments (line 162) | class FedKSeedTrainingArguments(KSeedTrainingArguments):

FILE: python/fate_llm/algo/fedkseed/optimizer.py
  class RandomWalkOptimizer (line 55) | class RandomWalkOptimizer(Optimizer):
    method __init__ (line 62) | def __init__(self, params, lr, weight_decay, grad_clip, defaults=None):
    method from_model (line 74) | def from_model(cls, model, lr, weight_decay, grad_clip, **kwargs):
    method directional_derivative_step (line 81) | def directional_derivative_step(
    method step (line 95) | def step(self, closure: Optional[Callable[[], float]] = None) -> Optio...
  class ZerothOrderOptimizer (line 102) | class ZerothOrderOptimizer(RandomWalkOptimizer):
    method __init__ (line 103) | def __init__(self, params, lr, eps, weight_decay, grad_clip):
    method zeroth_order_step (line 108) | def zeroth_order_step(
    method random_perturb_parameters (line 152) | def random_perturb_parameters(self, directional_derivative_seed: int, ...
  class KSeedZerothOrderOptimizer (line 176) | class KSeedZerothOrderOptimizer(ZerothOrderOptimizer):
    method __init__ (line 177) | def __init__(
    method sample (line 193) | def sample(self) -> int:
    method step (line 201) | def step(self, closure: Callable[[], torch.FloatTensor] = None) -> tor...
    method kseed_zeroth_order_step (line 210) | def kseed_zeroth_order_step(self, closure: Callable[[], torch.FloatTen...

FILE: python/fate_llm/algo/fedkseed/pytorch_utils.py
  function get_decay_parameter_names (line 7) | def get_decay_parameter_names(model) -> List[str]:
  function get_optimizer_parameters_grouped_with_decay (line 34) | def get_optimizer_parameters_grouped_with_decay(model, weight_decay: flo...

FILE: python/fate_llm/algo/fedkseed/trainer.py
  class KSeedZOExtendedTrainer (line 20) | class KSeedZOExtendedTrainer(Trainer):
    method __init__ (line 21) | def __init__(
    method configure_seed_candidates (line 55) | def configure_seed_candidates(self, seed_candidates: torch.LongTensor,...
    method get_directional_derivative_history (line 59) | def get_directional_derivative_history(self):
    method k_seed_zo_mode (line 71) | def k_seed_zo_mode(args):
    method training_step (line 74) | def training_step(self, model: nn.Module, inputs: Dict[str, Union[torc...
    method create_optimizer_and_scheduler (line 101) | def create_optimizer_and_scheduler(self, num_training_steps: int):

FILE: python/fate_llm/algo/fedkseed/zo_utils.py
  function probability_from_amps (line 6) | def probability_from_amps(amps: List[List[float]], clip):
  function directional_derivative_step (line 23) | def directional_derivative_step(
  function build_seed_candidates (line 57) | def build_seed_candidates(k, low=0, high=2**32):
  function get_even_seed_probabilities (line 64) | def get_even_seed_probabilities(k):

FILE: python/fate_llm/algo/fedmkt/fedmkt.py
  class FedMKTTrainingArguments (line 46) | class FedMKTTrainingArguments(Seq2SeqTrainingArguments):
    method to_dict (line 88) | def to_dict(self):
    method to_dict_without_extra_args (line 102) | def to_dict_without_extra_args(self):
    method to_dict_with_client_priv_training_args (line 128) | def to_dict_with_client_priv_training_args(self):
    method to_dict_with_client_kd_args (line 135) | def to_dict_with_client_kd_args(self):
    method to_dict_with_server_kd_args (line 142) | def to_dict_with_server_kd_args(self):
  class FedMKTBase (line 149) | class FedMKTBase(object):
    method __init__ (line 150) | def __init__(self, *args, **kwargs):
    method save_model (line 154) | def save_model(
  class FedMKTSLM (line 175) | class FedMKTSLM(FedMKTBase):
    method __init__ (line 176) | def __init__(
    method train (line 229) | def train(self):
    method _init_trainer_for_distill (line 314) | def _init_trainer_for_distill(self, train_set):
    method _get_priv_data_training_args (line 339) | def _get_priv_data_training_args(self):
    method _get_pub_data_kd_training_args (line 345) | def _get_pub_data_kd_training_args(self):
    method _init_aggregator (line 351) | def _init_aggregator(self, ctx: Context, fed_args: FedArguments):
  class FedMKTLLM (line 364) | class FedMKTLLM(FedMKTBase):
    method __init__ (line 365) | def __init__(
    method _init_aggregator (line 409) | def _init_aggregator(self, ctx: Context):
    method generate_pub_data_logits (line 414) | def generate_pub_data_logits(self, first_epoch=False):
    method on_epoch_begin (line 430) | def on_epoch_begin(self, iter_ctx, epoch_idx, previous_pub_dataset):
    method on_epoch_end (line 484) | def on_epoch_end(self, iter_ctx, epoch_idx):
    method _get_pub_data_kd_training_args (line 510) | def _get_pub_data_kd_training_args(self):
    method train (line 516) | def train(self):

FILE: python/fate_llm/algo/fedmkt/fedmkt_data_collator.py
  class DataCollatorForFedMKT (line 40) | class DataCollatorForFedMKT(DataCollatorForSeq2Seq):
    method __init__ (line 54) | def __init__(self, *args, **kwargs):
    method __call__ (line 66) | def __call__(self, features, return_tensors=None):

FILE: python/fate_llm/algo/fedmkt/fedmkt_trainer.py
  class FedMKTTrainer (line 36) | class FedMKTTrainer(Seq2SeqTrainer):
    method __init__ (line 45) | def __init__(self, *args, **kwargs):
    method compute_loss (line 56) | def compute_loss(self, model, inputs, return_outputs=False):

FILE: python/fate_llm/algo/fedmkt/token_alignment/token_align.py
  function dtw (line 40) | def dtw(series_1, series_2, norm_func=np.linalg.norm):
  function greedy_dynamic_matching (line 84) | def greedy_dynamic_matching(base_model_tokens, blending_model_tokens, ba...
  function align_blending_model_logits_with_base_model_logits (line 179) | def align_blending_model_logits_with_base_model_logits(base_examples,
  function transform_step_logits (line 231) | def transform_step_logits(base_model_tokenizer: transformers.tokenizatio...
  function token_align (line 353) | def token_align(

FILE: python/fate_llm/algo/fedmkt/token_alignment/vocab_mapping.py
  function find_best_mapping (line 32) | def find_best_mapping(x, base_tokens, blending_model_special_token, base...
  function get_vocab_mappings (line 47) | def get_vocab_mappings(model_name_or_path, candidate_model_name_or_path,...

FILE: python/fate_llm/algo/fedmkt/utils/dataset_sync_util.py
  function sync_dataset (line 29) | def sync_dataset(dataset, local_rank, world_size, device):

FILE: python/fate_llm/algo/fedmkt/utils/generate_logit_utils.py
  class Metric (line 26) | class Metric(object):
    method cal_metric (line 28) | def cal_metric(cls, logits, input_ids, attention_mask, labels, trainin...
    method cal_ce (line 35) | def cal_ce(cls, logits, input_ids, attention_mask, labels, training_ar...
  class LogitsSelection (line 44) | class LogitsSelection(object):
    method select_logits (line 46) | def select_logits(cls, logits, training_args):
    method select_highest (line 53) | def select_highest(cls, logits, top_k_logits_keep):
  function generate_pub_data_logits (line 60) | def generate_pub_data_logits(inputs, model, training_args, data_collator):

FILE: python/fate_llm/algo/fedmkt/utils/tokenizer_tool.py
  function get_vocab_size (line 19) | def get_vocab_size(tokenizer_name_or_path):

FILE: python/fate_llm/algo/inferdpt/_encode_decode.py
  class EncoderDecoder (line 25) | class EncoderDecoder(object):
    method __init__ (line 27) | def __init__(self, ctx: Context) -> None:
    method encode (line 30) | def encode(self,  docs: List[Dict[str, str]], format_template: str):
    method decode (line 33) | def decode(self,  docs: List[Dict[str, str]], format_template: str ):
    method inference (line 36) | def inference(self, docs: List[Dict[str, str]], inference_kwargs: dict...

FILE: python/fate_llm/algo/inferdpt/inferdpt.py
  class InferDPTClient (line 34) | class InferDPTClient(EncoderDecoder):
    method __init__ (line 36) | def __init__(self, ctx: Context, inferdpt_pertub_kit: InferDPTKit, loc...
    method encode (line 44) | def encode(self, docs: List[Dict[str, str]], format_template: str = No...
    method _remote_inference (line 64) | def _remote_inference(self, docs: List[Dict[str, str]],
    method decode (line 97) | def decode(self, p_docs: List[Dict[str, str]], instruction_template: s...
    method inference (line 115) | def inference(self, docs: Union[List[Dict[str, str]], HuggingfaceDatas...
  class InferDPTServer (line 149) | class InferDPTServer(object):
    method __init__ (line 151) | def __init__(self, ctx: Context, inference_inst: Inference) -> None:
    method inference (line 157) | def inference(self, verbose=False):
    method predict (line 170) | def predict(self):

FILE: python/fate_llm/algo/inferdpt/init/_init.py
  class InferInit (line 21) | class InferInit(object):
    method __init__ (line 23) | def __init__(self, ctx: Context):
    method get_inst (line 26) | def get_inst(self):

FILE: python/fate_llm/algo/inferdpt/init/default_init.py
  class InferDPTAPIClientInit (line 23) | class InferDPTAPIClientInit(InferInit):
    method __init__ (line 31) | def __init__(self, ctx):
    method get_inst (line 35) | def get_inst(self)-> InferDPTClient:
  class InferDPTAPIServerInit (line 42) | class InferDPTAPIServerInit(InferInit):
    method __init__ (line 48) | def __init__(self, ctx):
    method get_inst (line 52) | def get_inst(self)-> InferDPTServer:

FILE: python/fate_llm/algo/inferdpt/utils.py
  class NumpyEncoder (line 33) | class NumpyEncoder(json.JSONEncoder):
    method default (line 36) | def default(self, obj):
  function save_jsonl (line 46) | def save_jsonl(filename, data):
  function create_sensitivity_of_embeddings (line 53) | def create_sensitivity_of_embeddings(all_embedding_matrix):
  function create_sorted_embedding_matrix (line 64) | def create_sorted_embedding_matrix(token_list, similarity_matrix):
  function cosine_similarity_vectors (line 74) | def cosine_similarity_vectors(A, B):
  class InferDPTKit (line 82) | class InferDPTKit(object):
    method __init__ (line 84) | def __init__(self, token_to_vector_dict, sorted_similarities, delta_f,...
    method save_to_path (line 92) | def save_to_path(self, path):
    method make_inferdpt_kit_param (line 111) | def make_inferdpt_kit_param(embedding_matrix: np.ndarray, token_list: ...
    method load_from_path (line 132) | def load_from_path(path):
    method perturb (line 144) | def perturb(self, doc: str, epsilon: float) -> str:
  function cosine_similarity_vectors (line 184) | def cosine_similarity_vectors(A, B):
  function add_laplace_noise_to_vector (line 192) | def add_laplace_noise_to_vector(vector, epsilon, delta_f_new):
  function perturb_sentence (line 204) | def perturb_sentence(sent,

FILE: python/fate_llm/algo/offsite_tuning/offsite_tuning.py
  class OffsiteTuningTrainerClient (line 22) | class OffsiteTuningTrainerClient(Seq2SeqFedAVGClient):
    method __init__ (line 24) | def __init__(
    method _share_model (line 70) | def _share_model(self, model, args: Seq2SeqTrainingArguments, sync_tra...
    method on_train_begin (line 82) | def on_train_begin(self, ctx: Context, aggregator: Aggregator, fed_arg...
    method on_federation (line 102) | def on_federation(
    method on_train_end (line 120) | def on_train_end(self, ctx: Context, aggregator: Aggregator, fed_args:...
    method init_aggregator (line 132) | def init_aggregator(self, ctx: Context, fed_args: FedArguments):
  class OffsiteTuningTrainerServer (line 139) | class OffsiteTuningTrainerServer(Seq2SeqFedAVGServer):
    method __init__ (line 141) | def __init__(self, ctx: Context, model: OffsiteTuningBaseModel, aggreg...
    method on_train_begin (line 147) | def on_train_begin(self, ctx: Context, aggregator: Aggregator):
    method on_train_end (line 154) | def on_train_end(self, ctx: Context, aggregator: Aggregator):
    method on_federation (line 159) | def on_federation(self, ctx: Context, aggregator, agg_iter_idx: int):
    method init_aggregator (line 165) | def init_aggregator(self, ctx):
    method train (line 171) | def train(self):
    method save_model (line 182) | def save_model(

FILE: python/fate_llm/data/data_collator/cust_data_collator.py
  function get_data_collator (line 20) | def get_data_collator(data_collator_name,
  function get_seq2seq_data_collator (line 45) | def get_seq2seq_data_collator(tokenizer_name_or_path, **kwargs):

FILE: python/fate_llm/data/data_collator/fedcot_collator.py
  class PrefixDataCollator (line 5) | class PrefixDataCollator(DataCollatorForSeq2Seq):
    method __call__ (line 6) | def __call__(self, features, return_tensors=None):
  function get_prefix_data_collator (line 17) | def get_prefix_data_collator(tokenizer_name_or_path):

FILE: python/fate_llm/data/tokenizers/cust_tokenizer.py
  function get_tokenizer (line 19) | def get_tokenizer(

FILE: python/fate_llm/dataset/fedcot_dataset.py
  class PrefixDataset (line 12) | class PrefixDataset(InputOutputDataset):
    method __init__ (line 14) | def __init__(self,
    method load_rationale (line 30) | def load_rationale(self, result_list, key='rationale'):
    method get_str_item (line 34) | def get_str_item(self, i) -> dict:
    method get_tokenized_item (line 53) | def get_tokenized_item(self, i) -> dict:

FILE: python/fate_llm/dataset/flex_dataset.py
  function get_jinjax_placeholders (line 38) | def get_jinjax_placeholders(jinjax_text, placeholder_count=2):
  function regex_replace (line 45) | def regex_replace(string, pattern, repl, count: int = 0):
  function apply_template (line 62) | def apply_template(template, data):
  function tokenize_flex_dataset (line 77) | def tokenize_flex_dataset(raw_datasets, tokenizer, sub_domain, tokenize_...
  class FlexDataset (line 116) | class FlexDataset(Dataset):
    method __init__ (line 117) | def __init__(self,
    method parse_config (line 161) | def parse_config(self, config=None):
    method get_generate_prompt (line 176) | def get_generate_prompt(self, tokenize=True, return_tensors="pt"):
    method construct_prompt_list (line 187) | def construct_prompt_list(samples_dict, num_shot_per_label, prompt_num...
    method group_text_label_list (line 221) | def group_text_label_list(text_list, label_list):
    method prepare_few_shot (line 225) | def prepare_few_shot(self, text_list, label_list, aug_prompt_num):
    method prepare_augment (line 240) | def prepare_augment(self, text_list, label_list, aug_prompt_num):
    method abstract_from_augmented (line 250) | def abstract_from_augmented(self, sample_list):
    method prepare_query_to_filter_clustered (line 275) | def prepare_query_to_filter_clustered(self, clustered_sentences_list, ...
    method parse_clustered_response (line 288) | def parse_clustered_response(self, clustered_sentence, clustered_label...
    method group_data_list (line 315) | def group_data_list(data_list, text_key, label_key):
    method load (line 321) | def load(self, path):
    method apply_chat_template (line 341) | def apply_chat_template(self, query):
    method get_raw_dataset (line 358) | def get_raw_dataset(self):
    method __len__ (line 361) | def __len__(self):
    method get_item (line 364) | def get_item(self, i):
    method get_item_dict (line 367) | def get_item_dict(self, i):
    method __getitem__ (line 371) | def __getitem__(self, i) -> dict:

FILE: python/fate_llm/dataset/hf_dataset.py
  class HuggingfaceDataset (line 28) | class HuggingfaceDataset(Dataset):
    method __init__ (line 33) | def __init__(
    method load (line 92) | def load(self, file_path):
    method __getitem__ (line 113) | def __getitem__(self, idx):
    method __len__ (line 118) | def __len__(self):
  class Dolly15K (line 124) | class Dolly15K(HuggingfaceDataset):
    method __init__ (line 174) | def __init__(self, *args, **kwargs):
    method load (line 178) | def load(self, file_path):
    method _post_process (line 182) | def _post_process(self, dataset):

FILE: python/fate_llm/dataset/input_output_dataset.py
  class InputOutputDataset (line 12) | class InputOutputDataset(Dataset):
    method __init__ (line 14) | def __init__(self,
    method load (line 37) | def load(self, path):
    method get_raw_dataset (line 66) | def get_raw_dataset(self):
    method __len__ (line 69) | def __len__(self):
    method get_str_item (line 72) | def get_str_item(self, i) -> dict:
    method _process_item (line 82) | def _process_item(self, data_item):
    method get_tokenized_item (line 105) | def get_tokenized_item(self, i) -> dict:
    method __getitem__ (line 111) | def __getitem__(self, i) -> dict:

FILE: python/fate_llm/dataset/prompt_dataset.py
  class PromptDataset (line 28) | class PromptDataset(Dataset):
    method __init__ (line 29) | def __init__(self,
    method load (line 75) | def load(self, file_path):
    method _process_data (line 121) | def _process_data(examples, tokenizer, prompt_template, prompt_column,
    method _pad_to_max_length (line 172) | def _pad_to_max_length(examples, tokenizer, max_length):
    method get_vocab_size (line 196) | def get_vocab_size(self):
    method __getitem__ (line 199) | def __getitem__(self, item):
    method __len__ (line 202) | def __len__(self):
    method __repr__ (line 205) | def __repr__(self):

FILE: python/fate_llm/dataset/qa_dataset.py
  class PIQA (line 25) | class PIQA:
    method __init__ (line 26) | def __init__(self):
    method get_context (line 29) | def get_context(self, examples):
    method get_target (line 33) | def get_target(self, examples):
  class SciQ (line 42) | class SciQ:
    method __init__ (line 43) | def __init__(self):
    method get_context (line 46) | def get_context(self, examples):
    method get_target (line 51) | def get_target(self, examples):
  class OpenBookQA (line 55) | class OpenBookQA:
    method get_context (line 56) | def get_context(self, examples):
    method get_target (line 59) | def get_target(self, examples):
  class ARC (line 69) | class ARC:
    method __init__ (line 70) | def __init__(self):
    method get_context (line 73) | def get_context(self, examples):
    method get_target (line 77) | def get_target(self, examples):
  class WIC (line 88) | class WIC:
    method __init__ (line 89) | def __init__(self):
    method get_context (line 93) | def get_context(self, examples):
    method get_target (line 107) | def get_target(self, examples):
  class BoolQ (line 116) | class BoolQ:
    method __init__ (line 117) | def __init__(self):
    method get_context (line 120) | def get_context(self, examples):
    method get_target (line 127) | def get_target(self, examples):
  class CommonsenseQA (line 130) | class CommonsenseQA:
    method get_context (line 131) | def get_context(self, examples):
    method get_target (line 134) | def get_target(self, examples):
  class RTE (line 144) | class RTE:
    method __init__ (line 145) | def __init__(self):
    method get_context (line 148) | def get_context(self, examples):
    method get_target (line 159) | def get_target(self, examples):
  function tokenize_qa_dataset (line 177) | def tokenize_qa_dataset(dataset_name, tokenizer, save_path=None, seq_max...
  class QaDataset (line 268) | class QaDataset(Dataset):
    method __init__ (line 270) | def __init__(self,
    method load (line 296) | def load(self, path):
    method set_return_with_idx (line 317) | def set_return_with_idx(self):
    method reset_return_with_idx (line 320) | def reset_return_with_idx(self):
    method __len__ (line 323) | def __len__(self):
    method __getitem__ (line 326) | def __getitem__(self, idx):

FILE: python/fate_llm/dataset/seq_cls_dataset.py
  class SeqCLSDataset (line 27) | class SeqCLSDataset(Dataset):
    method __init__ (line 45) | def __init__(
    method load (line 74) | def load(self, file_path):
    method get_classes (line 97) | def get_classes(self):
    method get_vocab_size (line 100) | def get_vocab_size(self):
    method get_sample_ids (line 103) | def get_sample_ids(self):
    method __getitem__ (line 106) | def __getitem__(self, item):
    method __len__ (line 118) | def __len__(self):
    method __repr__ (line 121) | def __repr__(self):

FILE: python/fate_llm/evaluate/scripts/_options.py
  function parse_custom_type (line 9) | def parse_custom_type(value):
  class LlmSharedOptions (line 19) | class LlmSharedOptions(object):
    method __init__ (line 31) | def __init__(self):
    method __getitem__ (line 34) | def __getitem__(self, item):
    method get (line 37) | def get(self, k, default=None):
    method update (line 43) | def update(self, **kwargs):
    method post_process (line 48) | def post_process(self):
    method get_shared_options (line 61) | def get_shared_options(cls, hidden=False):

FILE: python/fate_llm/evaluate/scripts/config_cli.py
  function eval_config_group (line 26) | def eval_config_group():
  function _new (line 34) | def _new():
  function _edit (line 45) | def _edit(ctx, **kwargs):
  function _show (line 56) | def _show():

FILE: python/fate_llm/evaluate/scripts/data_cli.py
  function download_data (line 34) | def download_data(ctx, tasks, **kwargs):

FILE: python/fate_llm/evaluate/scripts/eval_cli.py
  function run_evaluate (line 41) | def run_evaluate(ctx, include, eval_config, result_output, **kwargs):
  function run_job_eval (line 70) | def run_job_eval(job, eval_conf):
  function run_suite_eval (line 102) | def run_suite_eval(suite, eval_conf, output_path=None):

FILE: python/fate_llm/evaluate/scripts/fate_llm_cli.py
  class FATELlmCLI (line 35) | class FATELlmCLI(click.MultiCommand):
    method list_commands (line 37) | def list_commands(self, ctx):
    method get_command (line 40) | def get_command(self, ctx, name):
  function fate_llm_cli (line 51) | def fate_llm_cli(ctx, **kwargs):

FILE: python/fate_llm/evaluate/tasks/__init__.py
  function local_fn_constructor (line 21) | def local_fn_constructor(loader, node):
  function local_fn_representer (line 25) | def local_fn_representer(dumper, data):
  function dump_yaml (line 29) | def dump_yaml(dict, path):
  class Task (line 34) | class Task:
    method task_name (line 42) | def task_name(self):
    method task_template (line 46) | def task_template(self):
    method task_scr_dir (line 53) | def task_scr_dir(self):
    method task_conf_path (line 57) | def task_conf_path(self):
    method task_source_url (line 61) | def task_source_url(self):
    method download_from_source (line 64) | def download_from_source(self):
  class Dolly (line 68) | class Dolly(Task):
    method download_from_source (line 73) | def download_from_source(self):
  class AdvertiseGen (line 85) | class AdvertiseGen(Task):
    method download_from_source (line 92) | def download_from_source(self):

FILE: python/fate_llm/evaluate/tasks/advertise_gen/advertise_utils.py
  function rouge_l (line 8) | def rouge_l(predictions, references, use_stemmer=False):

FILE: python/fate_llm/evaluate/tasks/dolly_15k/dolly_utils.py
  function rouge_l (line 7) | def rouge_l(predictions, references, use_stemmer=False):
  function doc_to_text (line 17) | def doc_to_text(doc):

FILE: python/fate_llm/evaluate/utils/_io.py
  class echo (line 21) | class echo(object):
    method set_file (line 25) | def set_file(cls, file):
    method echo (line 29) | def echo(cls, message, **kwargs):
    method sep_line (line 34) | def sep_line(cls):
    method file (line 38) | def file(cls, message, **kwargs):
    method stdout (line 42) | def stdout(cls, message, **kwargs):
    method stdout_newline (line 46) | def stdout_newline(cls):
    method welcome (line 50) | def welcome(cls):
    method flush (line 55) | def flush(cls):
  function set_logger (line 60) | def set_logger(name):

FILE: python/fate_llm/evaluate/utils/_parser.py
  class LlmJob (line 23) | class LlmJob(object):
    method __init__ (line 24) | def __init__(self, job_name: str, script_path: Path=None, conf_path: P...
  class LlmPair (line 43) | class LlmPair(object):
    method __init__ (line 44) | def __init__(
  class LlmSuite (line 51) | class LlmSuite(object):
    method __init__ (line 52) | def __init__(
    method load (line 61) | def load(path: Path):
    method update_status (line 131) | def update_status(
    method get_final_status (line 139) | def get_final_status(self):

FILE: python/fate_llm/evaluate/utils/config.py
  function create_eval_config (line 44) | def create_eval_config(path: Path, override=False):
  function default_eval_config (line 52) | def default_eval_config():
  class Config (line 58) | class Config(object):
    method __init__ (line 59) | def __init__(self, config):
    method update_conf (line 62) | def update_conf(self, **kwargs):
    method load (line 67) | def load(path: typing.Union[str, Path], **kwargs):
    method load_from_file (line 79) | def load_from_file(path: typing.Union[str, Path]):
  function parse_config (line 104) | def parse_config(config):
  function _set_namespace (line 112) | def _set_namespace(namespace):

FILE: python/fate_llm/evaluate/utils/data_tools.py
  function download_data (line 18) | def download_data(data_dir, data_url, is_tar=True):

FILE: python/fate_llm/evaluate/utils/llm_evaluator.py
  function evaluate (line 32) | def evaluate(tasks, model="hf", model_args=None, include_path=None, task...
  function aggregate_table (line 83) | def aggregate_table(results):
  function get_task_template (line 151) | def get_task_template(task):
  function export_config (line 159) | def export_config(config, task, export_dir=None, export_sub_dir=None):
  function copy_directory_to_dst (line 176) | def copy_directory_to_dst(src_dir, dst_dir, target_conf_file, new_conf: ...
  function contains_subdirectory (line 199) | def contains_subdirectory(path, subdirectories):
  function delete_config (line 211) | def delete_config(target_dir, force=False):
  function set_environ_fate_llm_base (line 221) | def set_environ_fate_llm_base(path):
  function set_environ_fate_llm_task_base (line 226) | def set_environ_fate_llm_task_base(path):
  function init_tasks (line 231) | def init_tasks(root_path=None):
  function download_task (line 274) | def download_task(tasks=None):

FILE: python/fate_llm/evaluate/utils/model_tools.py
  function load_model_from_path (line 22) | def load_model_from_path(model_path, peft_path=None, peft_config=None, m...
  function load_model (line 44) | def load_model(model_path, peft_path=None, model_args=None):
  function load_by_loader (line 49) | def load_by_loader(loader_name=None, loader_conf_path=None, peft_path=No...

FILE: python/fate_llm/inference/api.py
  class APICompletionInference (line 23) | class APICompletionInference(Inference):
    method __init__ (line 25) | def __init__(self, api_url: str, model_name: str, api_key: str = 'EMPT...
    method inference (line 34) | def inference(self, docs: List[str], inference_kwargs: dict = {}) -> L...

FILE: python/fate_llm/inference/hf_qw.py
  class QwenHFCompletionInference (line 23) | class QwenHFCompletionInference(Inference):
    method __init__ (line 25) | def __init__(self, model, tokenizer):
    method inference (line 29) | def inference(self, docs: List[str], inference_kwargs: dict = {}) -> L...

FILE: python/fate_llm/inference/inference_base.py
  class Inference (line 20) | class Inference(object):
    method __init__ (line 22) | def __init__(self):
    method inference (line 25) | def inference(self, docs: List[str], inference_kwargs: dict = {}) -> L...

FILE: python/fate_llm/inference/vllm.py
  class VLLMInference (line 27) | class VLLMInference(Inference):
    method __init__ (line 29) | def __init__(self, model_path, num_gpu=1, dtype='float16', gpu_memory_...
    method inference (line 34) | def inference(self, docs: List[str], inference_kwargs: dict = {}) -> L...

FILE: python/fate_llm/model_zoo/embedding_transformer/st_model.py
  class SentenceTransformerModel (line 20) | class SentenceTransformerModel(object):
    method __init__ (line 21) | def __init__(
    method load (line 53) | def load(self):

FILE: python/fate_llm/model_zoo/hf_model.py
  class HFAutoModelForCausalLM (line 20) | class HFAutoModelForCausalLM:
    method __init__ (line 22) | def __init__(self, pretrained_model_name_or_path, *model_args, **kwarg...
    method load (line 30) | def load(self):

FILE: python/fate_llm/model_zoo/offsite_tuning/bloom.py
  class BloomMainModel (line 23) | class BloomMainModel(OffsiteTuningMainModel):
    method __init__ (line 25) | def __init__(
    method get_base_model (line 38) | def get_base_model(self):
    method get_model_transformer_blocks (line 41) | def get_model_transformer_blocks(self, model: BloomForCausalLM):
    method get_additional_param_state_dict (line 44) | def get_additional_param_state_dict(self):
    method load_additional_param_state_dict (line 60) | def load_additional_param_state_dict(self, submodel_weights: dict):
    method forward (line 80) | def forward(
  class BloomSubModel (line 110) | class BloomSubModel(OffsiteTuningSubModel):
    method __init__ (line 112) | def __init__(
    method get_base_model (line 132) | def get_base_model(self):
    method get_model_transformer_blocks (line 140) | def get_model_transformer_blocks(self, model: BloomForCausalLM):
    method get_additional_param_state_dict (line 143) | def get_additional_param_state_dict(self):
    method load_additional_param_state_dict (line 159) | def load_additional_param_state_dict(self, submodel_weights: dict):
    method forward (line 179) | def forward(
    method parameters (line 208) | def parameters(self, recurse=True):

FILE: python/fate_llm/model_zoo/offsite_tuning/gpt2.py
  class GPT2LMHeadMainModel (line 22) | class GPT2LMHeadMainModel(OffsiteTuningMainModel):
    method __init__ (line 24) | def __init__(
    method get_base_model (line 37) | def get_base_model(self):
    method get_model_transformer_blocks (line 40) | def get_model_transformer_blocks(self, model: GPT2LMHeadModel):
    method forward (line 43) | def forward(self,
    method get_additional_param_state_dict (line 75) | def get_additional_param_state_dict(self):
    method load_additional_param_state_dict (line 94) | def load_additional_param_state_dict(self, submodel_weights: dict):
  class GPT2LMHeadSubModel (line 119) | class GPT2LMHeadSubModel(OffsiteTuningSubModel):
    method __init__ (line 121) | def __init__(
    method get_base_model (line 141) | def get_base_model(self):
    method get_model_transformer_blocks (line 149) | def get_model_transformer_blocks(self, model: GPT2LMHeadModel):
    method get_additional_param_state_dict (line 152) | def get_additional_param_state_dict(self):
    method load_additional_param_state_dict (line 171) | def load_additional_param_state_dict(self, submodel_weights: dict):
    method forward (line 195) | def forward(self,
    method parameters (line 227) | def parameters(self, recurse=True):

FILE: python/fate_llm/model_zoo/offsite_tuning/llama.py
  class LlamaMainModel (line 20) | class LlamaMainModel(OffsiteTuningMainModel):
    method __init__ (line 22) | def __init__(
    method get_base_model (line 35) | def get_base_model(self):
    method get_model_transformer_blocks (line 38) | def get_model_transformer_blocks(self, model: LlamaForCausalLM):
    method get_additional_param_state_dict (line 41) | def get_additional_param_state_dict(self):
    method load_additional_param_state_dict (line 56) | def load_additional_param_state_dict(self, submodel_weights: dict):
    method forward (line 74) | def forward(self, **kwargs):
  class LlamaSubModel (line 78) | class LlamaSubModel(OffsiteTuningSubModel):
    method __init__ (line 80) | def __init__(
    method get_base_model (line 100) | def get_base_model(self):
    method get_model_transformer_blocks (line 108) | def get_model_transformer_blocks(self, model: LlamaForCausalLM):
    method get_additional_param_state_dict (line 111) | def get_additional_param_state_dict(self):
    method load_additional_param_state_dict (line 126) | def load_additional_param_state_dict(self, submodel_weights: dict):
    method forward (line 144) | def forward(self, **kwargs):
    method parameters (line 147) | def parameters(self, recurse=True):

FILE: python/fate_llm/model_zoo/offsite_tuning/offsite_tuning_model.py
  function get_dropout_emulator_and_adapters (line 25) | def get_dropout_emulator_and_adapters(
  function split_numpy_array (line 61) | def split_numpy_array(embedding_matrix, n, suffix):
  function recover_numpy_array (line 74) | def recover_numpy_array(slices_dict, suffix=""):
  class OffsiteTuningBaseModel (line 81) | class OffsiteTuningBaseModel(t.nn.Module):
    method __init__ (line 83) | def __init__(self, emulator_layer_num: int, adapter_top_layer_num: int...
    method initialize_model (line 97) | def initialize_model(self):
    method post_initialization (line 103) | def post_initialization(self):
    method get_adapter_top (line 106) | def get_adapter_top(self):
    method get_adapter_bottom (line 109) | def get_adapter_bottom(self):
    method get_emulator (line 112) | def get_emulator(self):
    method get_additional_param_state_dict (line 115) | def get_additional_param_state_dict(self):
    method load_additional_param_state_dict (line 119) | def load_additional_param_state_dict(self, submodel_weights: dict):
    method _get_numpy_arr (line 123) | def _get_numpy_arr(self, v):
    method load_numpy_state_dict (line 133) | def load_numpy_state_dict(self, module_dict, state_dict):
    method get_numpy_state_dict (line 144) | def get_numpy_state_dict(self, module_dict):
    method get_submodel_weights (line 153) | def get_submodel_weights(self, with_emulator=True) -> dict:
    method load_submodel_weights (line 177) | def load_submodel_weights(self, submodel_weights: dict, with_emulator=...
    method forward (line 201) | def forward(self, **kwargs):
    method get_base_model (line 204) | def get_base_model(self):
    method get_model_transformer_blocks (line 207) | def get_model_transformer_blocks(self, model: t.nn.Module):
  class OffsiteTuningMainModel (line 211) | class OffsiteTuningMainModel(OffsiteTuningBaseModel):
    method post_initialization (line 213) | def post_initialization(self):
  class OffsiteTuningSubModel (line 217) | class OffsiteTuningSubModel(OffsiteTuningBaseModel):
    method post_initialization (line 219) | def post_initialization(self):

FILE: python/fate_llm/model_zoo/pellm/albert.py
  class Albert (line 21) | class Albert(PELLM):
    method __init__ (line 26) | def __init__(self, config: dict = None,
    method check_config (line 44) | def check_config(self, pretain_path):

FILE: python/fate_llm/model_zoo/pellm/bart.py
  class Bart (line 21) | class Bart(PELLM):
    method __init__ (line 25) | def __init__(self, config: dict = None,
    method check_config (line 42) | def check_config(self, pretrain_path):

FILE: python/fate_llm/model_zoo/pellm/bert.py
  class Bert (line 21) | class Bert(PELLM):
    method __init__ (line 25) | def __init__(self, config: dict = None,
    method check_config (line 42) | def check_config(self, pretrain_path):

FILE: python/fate_llm/model_zoo/pellm/bloom.py
  class Bloom (line 21) | class Bloom(PELLM):
    method __init__ (line 26) | def __init__(self, config: dict = None,

FILE: python/fate_llm/model_zoo/pellm/chatglm.py
  class ChatGLM (line 20) | class ChatGLM(PELLM):
    method __init__ (line 21) | def __init__(self,
    method init_config (line 38) | def init_config(self):
    method add_peft (line 44) | def add_peft(self):

FILE: python/fate_llm/model_zoo/pellm/deberta.py
  class Deberta (line 21) | class Deberta(PELLM):
    method __init__ (line 26) | def __init__(self, config: dict = None,
    method check_config (line 43) | def check_config(self, pretrain_path):

FILE: python/fate_llm/model_zoo/pellm/distilbert.py
  class DistilBert (line 21) | class DistilBert(PELLM):
    method __init__ (line 25) | def __init__(self, config: dict = None,
    method check_config (line 42) | def check_config(self, pretrain_path):

FILE: python/fate_llm/model_zoo/pellm/gpt2.py
  class GPT2 (line 21) | class GPT2(PELLM):
    method __init__ (line 25) | def __init__(self,
    method check_config (line 43) | def check_config(self, pretrain_path):
  class GPT2CLM (line 50) | class GPT2CLM(GPT2):

FILE: python/fate_llm/model_zoo/pellm/llama.py
  class LLaMa (line 22) | class LLaMa(PELLM):
    method __init__ (line 25) | def __init__(self,
    method init_base_lm (line 36) | def init_base_lm(self, **kwargs):
    method check_config (line 48) | def check_config(self, pretrain_path):

FILE: python/fate_llm/model_zoo/pellm/opt.py
  class OPT (line 21) | class OPT(PELLM):
    method __init__ (line 26) | def __init__(self, config: dict = None,

FILE: python/fate_llm/model_zoo/pellm/parameter_efficient_llm.py
  class PELLM (line 36) | class PELLM(torch.nn.Module):
    method __init__ (line 41) | def __init__(self,
    method _init_pelm (line 64) | def _init_pelm(self, **kwargs):
    method init_lm_with_peft (line 68) | def init_lm_with_peft(self, **kwargs):
    method init_config (line 73) | def init_config(self, **kwargs):
    method init_base_lm (line 86) | def init_base_lm(self, **kwargs):
    method add_peft (line 102) | def add_peft(self):
    method model_summary (line 116) | def model_summary(self):
    method forward (line 121) | def forward(self, *args, **kwargs):
    method save_trainable (line 129) | def save_trainable(self, output_path):
  class AutoPELLM (line 133) | class AutoPELLM(PELLM):
    method __init__ (line 135) | def __init__(self, **kwargs) -> None:

FILE: python/fate_llm/model_zoo/pellm/qwen.py
  class Qwen (line 21) | class Qwen(PELLM):
    method __init__ (line 26) | def __init__(self, config: dict = None,

FILE: python/fate_llm/model_zoo/pellm/roberta.py
  class Roberta (line 21) | class Roberta(PELLM):
    method __init__ (line 25) | def __init__(self, config: dict = None,
    method check_config (line 42) | def check_config(self, pretrain_path):

FILE: python/fate_llm/runner/fdkt_runner.py
  class FDKTRunner (line 37) | class FDKTRunner(DefaultRunner):
    method __init__ (line 38) | def __init__(
    method common_setup (line 73) | def common_setup(self, saved_model=None, output_dir=None):
    method llm_setup (line 112) | def llm_setup(self, train_set=None, validate_set=None, output_dir=None...
    method slm_setup (line 140) | def slm_setup(self, train_set=None, validate_set=None, output_dir=None...
    method train (line 169) | def train(
    method predict (line 195) | def predict(self, *args, **kwargs):

FILE: python/fate_llm/runner/fedcot_runner.py
  function _check_instances (line 43) | def _check_instances(
  class FedCoTRunner (line 68) | class FedCoTRunner(NNRunner):
    method __init__ (line 69) | def __init__(
    method _get_infer_inst (line 111) | def _get_infer_inst(self, init_conf):
    method _prepare_data (line 121) | def _prepare_data(self, data, data_name):
    method client_setup (line 153) | def client_setup(self, train_set=None, validate_set=None, output_dir=N...
    method server_setup (line 225) | def server_setup(self, stage="train"):
    method train (line 232) | def train(
    method predict (line 267) | def predict(self, test_data: Union[str], saved_model_path: str = None)...

FILE: python/fate_llm/runner/fedkseed_runner.py
  class FedKSeedRunner (line 38) | class FedKSeedRunner(DefaultRunner):
    method __init__ (line 39) | def __init__(
    method client_setup (line 77) | def client_setup(self, train_set=None, validate_set=None, output_dir=N...
    method server_setup (line 113) | def server_setup(self, stage="train"):
  function maybe_loader_load_from_conf (line 127) | def maybe_loader_load_from_conf(conf):

FILE: python/fate_llm/runner/fedmkt_runner.py
  class FedMKTRunner (line 35) | class FedMKTRunner(DefaultRunner):
    method __init__ (line 37) | def __init__(
    method common_setup (line 83) | def common_setup(self, saved_model=None, output_dir=None):
    method llm_setup (line 134) | def llm_setup(self, train_set=None, validate_set=None, output_dir=None...
    method slm_setup (line 169) | def slm_setup(self, train_set=None, validate_set=None, output_dir=None...
    method train (line 206) | def train(
    method predict (line 230) | def predict(self, *args, **kwargs):

FILE: python/fate_llm/runner/homo_seq2seq_runner.py
  function _check_instances (line 44) | def _check_instances(
  class Seq2SeqRunner (line 81) | class Seq2SeqRunner(DefaultRunner):
    method __init__ (line 82) | def __init__(
    method client_setup (line 120) | def client_setup(self, train_set=None, validate_set=None, output_dir=N...
    method server_setup (line 192) | def server_setup(self, stage="train"):
    method predict (line 204) | def predict(self, test_data: Union[str, DataFrame], saved_model_path: ...

FILE: python/fate_llm/runner/inferdpt_runner.py
  class InferDPTRunner (line 45) | class InferDPTRunner(NNRunner):
    method __init__ (line 47) | def __init__(
    method _get_inst (line 71) | def _get_inst(self):
    method client_setup (line 79) | def client_setup(self):
    method server_setup (line 84) | def server_setup(self):
    method _prepare_data (line 89) | def _prepare_data(self, data, data_name):
    method train (line 115) | def train(
    method predict (line 145) | def predict(

FILE: python/fate_llm/runner/offsite_tuning_runner.py
  class OTRunner (line 41) | class OTRunner(Seq2SeqRunner):
    method __init__ (line 43) | def __init__(
    method setup (line 65) | def setup(self, train_set=None, validate_set=None, output_dir=None, sa...
    method server_setup (line 144) | def server_setup(self, stage="train"):
    method train (line 157) | def train(

FILE: python/fate_llm/trainer/seq2seq_trainer.py
  class _S2STrainingArguments (line 43) | class _S2STrainingArguments(_hf_Seq2SeqTrainingArguments):
    method __post_init__ (line 60) | def __post_init__(self):
  class Seq2SeqTrainingArguments (line 75) | class Seq2SeqTrainingArguments(_S2STrainingArguments):
    method to_dict (line 78) | def to_dict(self):
  class HomoSeq2SeqTrainerClient (line 88) | class HomoSeq2SeqTrainerClient(Seq2SeqTrainer, HomoTrainerMixin):
    method __init__ (line 90) | def __init__(
    method _save (line 150) | def _save(