LICENSE
README.md
pyproject.toml
fusion_bench/__init__.py
fusion_bench/__main__.py
fusion_bench.egg-info/PKG-INFO
fusion_bench.egg-info/SOURCES.txt
fusion_bench.egg-info/dependency_links.txt
fusion_bench.egg-info/entry_points.txt
fusion_bench.egg-info/requires.txt
fusion_bench.egg-info/top_level.txt
fusion_bench/../fusion_bench_config/README.md
fusion_bench/../fusion_bench_config/clip-vit-base-patch32_robustness_corrupted.yaml
fusion_bench/../fusion_bench_config/fabric_model_fusion.yaml
fusion_bench/../fusion_bench_config/llama_full_finetune.yaml
fusion_bench/../fusion_bench_config/llama_magnitude_pruning.yaml
fusion_bench/../fusion_bench_config/llama_model_fusion.yaml
fusion_bench/../fusion_bench_config/nyuv2_config.yaml
fusion_bench/../fusion_bench_config/nyuv2_mtl_train.yaml
fusion_bench/../fusion_bench_config/_get_started/clip_evaluate_single_model.yaml
fusion_bench/../fusion_bench_config/_get_started/clip_simple_average.yaml
fusion_bench/../fusion_bench_config/_get_started/clip_task_arithmetic.yaml
fusion_bench/../fusion_bench_config/_get_started/greeting_program.yaml
fusion_bench/../fusion_bench_config/_get_started/llm_slerp.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/README.md
fusion_bench/../fusion_bench_config/dataset/image_classification/test/TALL10.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/TALL12.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/TALL14.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/TALL16.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/TALL18.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/TALL20.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/cifar10.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/cifar100.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/cub-200-2011.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/dtd.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/emnist_letters.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/emnist_mnist.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/eurosat.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/fashion_mnist.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/fer2013.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/food101.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/gtsrb.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/kmnist.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/mango-leaf-disease.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/mnist.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/oxford-iiit-pet.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/oxford_flowers102.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/pcam.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/rendered-sst2.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/resisc45.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/stanford-cars.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/stl10.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/sun397.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/svhn.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/the_eight_tasks.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/test/tiny-imagenet.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/TALL10.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/TALL12.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/TALL14.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/TALL16.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/TALL18.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/TALL20.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/cifar10.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/cifar100.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/cub-200-2011.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/dtd.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/emnist_letters.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/emnist_mnist.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/eurosat.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/fashion_mnist.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/fer2013.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/food101.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/gtsrb.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/kmnist.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/mango-leaf-disease.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/mnist.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/oxford-iiit-pet.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/oxford_flowers102.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/pcam.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/rendered-sst2.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/resisc45.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/stanford-cars.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/stl10.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/sun397.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/svhn.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/the_eight_tasks.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/train/tiny-imagenet.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/val/dtd.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/val/eurosat.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/val/gtsrb.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/val/mnist.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/val/resisc45.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/val/stanford-cars.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/val/sun397.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/val/svhn.yaml
fusion_bench/../fusion_bench_config/dataset/image_classification/val/the_eight_tasks.yaml
fusion_bench/../fusion_bench_config/dataset/llm_sft/alpaca_cleaned.yaml
fusion_bench/../fusion_bench_config/dataset/llm_sft/ultrachat_200k.yaml
fusion_bench/../fusion_bench_config/dataset/question_answering/search_qa.yaml
fusion_bench/../fusion_bench_config/dataset/question_answering/test/search_qa.yaml
fusion_bench/../fusion_bench_config/dataset/question_answering/train/MetaMathQA.yaml
fusion_bench/../fusion_bench_config/dataset/question_answering/train/search_qa.yaml
fusion_bench/../fusion_bench_config/dataset/question_answering/val/search_qa.yaml
fusion_bench/../fusion_bench_config/dataset/summarization/xsum.yaml
fusion_bench/../fusion_bench_config/dataset/summarization/test/xsum.yaml
fusion_bench/../fusion_bench_config/dataset/summarization/train/xsum.yaml
fusion_bench/../fusion_bench_config/dataset/summarization/val/xsum.yaml
fusion_bench/../fusion_bench_config/dataset/text_generation/test/gsm-hard.yaml
fusion_bench/../fusion_bench_config/dataset/text_generation/test/gsm8k.yaml
fusion_bench/../fusion_bench_config/dataset/text_generation/test/gsm8k_question_label.yaml
fusion_bench/../fusion_bench_config/dataset/text_generation/train/CodeAlpaca-20k.yaml
fusion_bench/../fusion_bench_config/dataset/text_generation/train/gsm8k.yaml
fusion_bench/../fusion_bench_config/dataset/text_generation/train/gsm8k_question_label.yaml
fusion_bench/../fusion_bench_config/fabric/auto.yaml
fusion_bench/../fusion_bench_config/fabric/llama_ddp.yaml
fusion_bench/../fusion_bench_config/fabric/llama_fsdp.yaml
fusion_bench/../fusion_bench_config/fabric/llama_peft_fsdp.yaml
fusion_bench/../fusion_bench_config/fabric/loggers/csv_logger.yaml
fusion_bench/../fusion_bench_config/fabric/loggers/mlflow_logger.yaml
fusion_bench/../fusion_bench_config/fabric/loggers/tensorboard_logger.yaml
fusion_bench/../fusion_bench_config/fabric/loggers/wandb_logger.yaml
fusion_bench/../fusion_bench_config/fabric/strategy/deepspeed.yaml
fusion_bench/../fusion_bench_config/fabric/strategy/llama_fsdp.yaml
fusion_bench/../fusion_bench_config/fabric/strategy/llama_peft_fsdp.yaml
fusion_bench/../fusion_bench_config/hydra/default.yaml
fusion_bench/../fusion_bench_config/hydra/help/fusion_bench_help.yaml
fusion_bench/../fusion_bench_config/hydra/job_logging/rich_logging.yaml
fusion_bench/../fusion_bench_config/method/clip_finetune.yaml
fusion_bench/../fusion_bench_config/method/depth_upscaling.yaml
fusion_bench/../fusion_bench_config/method/dummy.yaml
fusion_bench/../fusion_bench_config/method/mixtral_moe_merging.yaml
fusion_bench/../fusion_bench_config/method/mixtral_moe_upscaling.yaml
fusion_bench/../fusion_bench_config/method/model_recombination.yaml
fusion_bench/../fusion_bench_config/method/simple_average.yaml
fusion_bench/../fusion_bench_config/method/task_arithmetic.yaml
fusion_bench/../fusion_bench_config/method/ties_merging.yaml
fusion_bench/../fusion_bench_config/method/ada_svd/clip_vision.yaml
fusion_bench/../fusion_bench_config/method/adamerging/clip.yaml
fusion_bench/../fusion_bench_config/method/adamerging/layer_wise_flan_t5.yaml
fusion_bench/../fusion_bench_config/method/adamerging/layer_wise_gpt2.yaml
fusion_bench/../fusion_bench_config/method/adamerging/llama_sft.yaml
fusion_bench/../fusion_bench_config/method/analysis/task_vector_cos_similarity.yaml
fusion_bench/../fusion_bench_config/method/analysis/task_vector_violin_plot.yaml
fusion_bench/../fusion_bench_config/method/bitdelta/bitdelta.yaml
fusion_bench/../fusion_bench_config/method/classification/clip_continual_finetune.yaml
fusion_bench/../fusion_bench_config/method/classification/clip_finetune.yaml
fusion_bench/../fusion_bench_config/method/concrete_subspace/clip_concrete_layer_wise_adamerging.yaml
fusion_bench/../fusion_bench_config/method/concrete_subspace/clip_concrete_task_arithmetic.yaml
fusion_bench/../fusion_bench_config/method/concrete_subspace/clip_concrete_task_wise_adamerging.yaml
fusion_bench/../fusion_bench_config/method/concrete_subspace/clip_post_defense_AWM.yaml
fusion_bench/../fusion_bench_config/method/concrete_subspace/clip_post_defense_SAU.yaml
fusion_bench/../fusion_bench_config/method/concrete_subspace/clip_safe_concrete_layer_wise_adamerging.yaml
fusion_bench/../fusion_bench_config/method/concrete_subspace/clip_safe_concrete_task_arithmetic.yaml
fusion_bench/../fusion_bench_config/method/dare/simple_average.yaml
fusion_bench/../fusion_bench_config/method/dare/task_arithmetic.yaml
fusion_bench/../fusion_bench_config/method/dare/ties_merging.yaml
fusion_bench/../fusion_bench_config/method/dawe/dawe_for_clip.yaml
fusion_bench/../fusion_bench_config/method/doge_ta/doge_ta.yaml
fusion_bench/../fusion_bench_config/method/ensemble/max_model_predictor.yaml
fusion_bench/../fusion_bench_config/method/ensemble/simple_ensemble.yaml
fusion_bench/../fusion_bench_config/method/ensemble/weighted_ensemble.yaml
fusion_bench/../fusion_bench_config/method/expert_sparsity/README.md
fusion_bench/../fusion_bench_config/method/expert_sparsity/mixtral.yaml
fusion_bench/../fusion_bench_config/method/fisher_merging/clip_fisher_merging.yaml
fusion_bench/../fusion_bench_config/method/fisher_merging/fisher_merging.yaml
fusion_bench/../fusion_bench_config/method/fisher_merging/gpt2_fisher_merging.yaml
fusion_bench/../fusion_bench_config/method/fw_merging/fw_hard.yaml
fusion_bench/../fusion_bench_config/method/fw_merging/fw_soft.yaml
fusion_bench/../fusion_bench_config/method/gossip/layer_wise_clip.yaml
fusion_bench/../fusion_bench_config/method/gossip/layer_wise_flan_t5.yaml
fusion_bench/../fusion_bench_config/method/isotropic_merging/iso_c.yaml
fusion_bench/../fusion_bench_config/method/isotropic_merging/iso_cts.yaml
fusion_bench/../fusion_bench_config/method/linear/expo.yaml
fusion_bench/../fusion_bench_config/method/linear/linear_interpolation.yaml
fusion_bench/../fusion_bench_config/method/linear/llama_expo.yaml
fusion_bench/../fusion_bench_config/method/linear/llama_expo_with_dare.yaml
fusion_bench/../fusion_bench_config/method/linear/simple_average_for_llama.yaml
fusion_bench/../fusion_bench_config/method/linear/task_arithmetic_for_llama.yaml
fusion_bench/../fusion_bench_config/method/linear/weighted_average.yaml
fusion_bench/../fusion_bench_config/method/linear/weighted_average_for_llama.yaml
fusion_bench/../fusion_bench_config/method/lm_finetune/bradley_terry_rm.yaml
fusion_bench/../fusion_bench_config/method/lm_finetune/fullfinetune_sft.yaml
fusion_bench/../fusion_bench_config/method/lm_finetune/peftfinetune_sft.yaml
fusion_bench/../fusion_bench_config/method/model_stock/model_stock.yaml
fusion_bench/../fusion_bench_config/method/moe_pruner/moe_pruner.yaml
fusion_bench/../fusion_bench_config/method/opcm/opcm.yaml
fusion_bench/../fusion_bench_config/method/opcm/task_arithmetic.yaml
fusion_bench/../fusion_bench_config/method/opcm/ties_merging.yaml
fusion_bench/../fusion_bench_config/method/opcm/weight_average.yaml
fusion_bench/../fusion_bench_config/method/pruning/llama_magnitude_pruning.yaml
fusion_bench/../fusion_bench_config/method/pruning/llama_random_pruning.yaml
fusion_bench/../fusion_bench_config/method/pruning/llama_sparsegpt_pruning.yaml
fusion_bench/../fusion_bench_config/method/pruning/llama_wanda_pruning.yaml
fusion_bench/../fusion_bench_config/method/pruning/magnitude_diff_pruning.yaml
fusion_bench/../fusion_bench_config/method/pwe_moe/epo_for_openclip.yaml
fusion_bench/../fusion_bench_config/method/pwe_moe/ls_for_openclip.yaml
fusion_bench/../fusion_bench_config/method/pwe_moe/pwe_moe_ls_for_clip.yaml
fusion_bench/../fusion_bench_config/method/randes/superposed_model_soup.yaml
fusion_bench/../fusion_bench_config/method/randes/superposed_task_arithmetic.yaml
fusion_bench/../fusion_bench_config/method/randes/superposed_task_arithmetic_lora.yaml
fusion_bench/../fusion_bench_config/method/rankone_moe/rankone_moe.yaml
fusion_bench/../fusion_bench_config/method/regmean/clip_regmean.yaml
fusion_bench/../fusion_bench_config/method/regmean/gpt2_regmean.yaml
fusion_bench/../fusion_bench_config/method/regmean/regmean.yaml
fusion_bench/../fusion_bench_config/method/regmean_plusplus/clip_regmean_plusplus.yaml
fusion_bench/../fusion_bench_config/method/slerp/slerp.yaml
fusion_bench/../fusion_bench_config/method/slerp/slerp_lm.yaml
fusion_bench/../fusion_bench_config/method/smile_upscaling/causal_lm_upscaling.yaml
fusion_bench/../fusion_bench_config/method/smile_upscaling/error_accumulation.yaml
fusion_bench/../fusion_bench_config/method/smile_upscaling/projected_energy.yaml
fusion_bench/../fusion_bench_config/method/smile_upscaling/singular_projection_merging.yaml
fusion_bench/../fusion_bench_config/method/smile_upscaling/smile_mistral_upscaling.yaml
fusion_bench/../fusion_bench_config/method/smile_upscaling/smile_qwen2_upscaling.yaml
fusion_bench/../fusion_bench_config/method/smile_upscaling/smile_upscaling.yaml
fusion_bench/../fusion_bench_config/method/sparselo_pruning/llama_iterative_sparselo.yaml
fusion_bench/../fusion_bench_config/method/sparselo_pruning/llama_pcp_sparselo.yaml
fusion_bench/../fusion_bench_config/method/sparselo_pruning/llama_sparselo.yaml
fusion_bench/../fusion_bench_config/method/surgery/adamerging_surgery.yaml
fusion_bench/../fusion_bench_config/method/tall_mask/task_arithmetic.yaml
fusion_bench/../fusion_bench_config/method/task_singular_vector/TaskSingularVectorMerging.yaml
fusion_bench/../fusion_bench_config/method/trust_region/clip_task_arithmetic.yaml
fusion_bench/../fusion_bench_config/method/wemoe/flan_t5_weight_ensembling_moe.yaml
fusion_bench/../fusion_bench_config/method/wemoe/sparse_weight_ensembling_moe.yaml
fusion_bench/../fusion_bench_config/method/wemoe/weight_ensembling_moe.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/README.md
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_TALL14.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_TALL20.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_cifar10.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_cifar100.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_dtd.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_eight_tasks.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_emnist_letters.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_eurosat.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_fashion_mnist.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_fer2013.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_food101.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_gtsrb.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_kmnist.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_mnist.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_oxford-iiit-pet.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_oxford_flowers102.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_pcam.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_rendered-sst2.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_resisc45.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_stanford-cars.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_stl10.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_sun397.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch16_svhn.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_TALL10.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_TALL12.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_TALL14.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_TALL16.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_TALL18.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_TALL20.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_cifar10.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_cifar100.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_dtd.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_eight_tasks.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_emnist_letters.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_eurosat.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_fashion_mnist.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_fer2013.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_food101.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_gtsrb.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_kmnist.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_mnist.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_oxford-iiit-pet.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_oxford_flowers102.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_pcam.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_rendered-sst2.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_resisc45.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_stanford-cars.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_stl10.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_sun397.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-base-patch32_svhn.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_TALL14.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_TALL20.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_cifar10.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_cifar100.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_dtd.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_eight_tasks.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_emnist_letters.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_eurosat.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_fashion_mnist.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_fer2013.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_food101.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_gtsrb.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_kmnist.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_mnist.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_oxford-iiit-pet.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_oxford_flowers102.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_pcam.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_rendered-sst2.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_resisc45.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_stanford-cars.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_stl10.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_sun397.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/clip-vit-large-patch14_svhn.yaml
fusion_bench/../fusion_bench_config/model/clip-vit/download_TALL20_models.sh
fusion_bench/../fusion_bench_config/model/clip-vit/generate_vit_model_config.sh
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-cola.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-cola_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-mnli.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-mnli_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-mrpc.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-mrpc_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-qnli.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-qnli_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-qqp.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-qqp_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-rte.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-rte_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-sst2.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-sst2_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-stsb.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-base_glue-stsb_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-large.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-large_glue-cola_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-large_glue-mnli_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-large_glue-mrpc_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-large_glue-qnli_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-large_glue-qqp_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-large_glue-rte_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-large_glue-sst2_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/flan-t5-large_glue-stsb_lora-16.yaml
fusion_bench/../fusion_bench_config/model/flan-t5/generate_flan-t5.sh
fusion_bench/../fusion_bench_config/modelpool/automodelpool.yaml
fusion_bench/../fusion_bench_config/modelpool/gpt-2_glue.yaml
fusion_bench/../fusion_bench_config/modelpool/nyuv2_modelpool.yaml
fusion_bench/../fusion_bench_config/modelpool/smile_mistral_exp_v1.yaml
fusion_bench/../fusion_bench_config/modelpool/smile_mistral_exp_v2.yaml
fusion_bench/../fusion_bench_config/modelpool/smile_mistral_exp_v3.yaml
fusion_bench/../fusion_bench_config/modelpool/smile_mistral_exp_v4.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/_template.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch16_TA8.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch16_TA8_lora.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch16_TA8_model_only.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch16_TALL14.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch16_TALL14_model_only.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch16_TALL20.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch16_TALL20_model_only.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch16_individual.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch16_individual_lora.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_TA8.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_TA8_control_task.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_TA8_model_only.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_TALL10.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_TALL12.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_TALL14.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_TALL14_model_only.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_TALL16.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_TALL18.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_TALL20.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_TALL20_model_only.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_cars_and_dtd.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_generalization_exp1.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_generalization_exp2.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_individual.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_mtl.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_robustness_clean.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_robustness_corrupted.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_single_finetuned.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_single_task_projection.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_sun397_and_cars.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_sun397_and_dtd.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_sun397_cars_and_dtd.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_svhn_and_mnist.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-base-patch32_two_tasks_control_task.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-large-patch14_TA8.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-large-patch14_TA8_model_only.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-large-patch14_TALL14.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-large-patch14_TALL14_model_only.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-large-patch14_TALL20.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-large-patch14_TALL20_model_only.yaml
fusion_bench/../fusion_bench_config/modelpool/CLIPVisionModelPool/clip-vit-large-patch14_individual.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/Qwen2.5-1.5B_math_and_coder.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/Qwen2.5-7B-math_and_coder.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/deepseek-v2-lite.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/llama_alpaca_cleaned.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/llama_codealpaca.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/llama_for_causallm.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/llama_metamathqa.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/llama_ultrachat.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/mistral-7b.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/mixtral-8x7b.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/mixtral_moe_merging.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/qwen2_math_1.5B_and_R1.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/simle_mixtral_exp_v4.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/single_llama_model.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/vicuna-7b-v1.5.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/mergebench/Llama-3.1-8B-Instruct.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/mergebench/Llama-3.1-8B.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/mergebench/Llama-3.2-3B-Instruct.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/mergebench/Llama-3.2-3B.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/mergebench/gemma-2-2b-it.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/mergebench/gemma-2-2b.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/mergebench/gemma-2-9b-it.yaml
fusion_bench/../fusion_bench_config/modelpool/CausalLMPool/mergebench/gemma-2-9b.yaml
fusion_bench/../fusion_bench_config/modelpool/OpenCLIPVisionModelPool/README.md
fusion_bench/../fusion_bench_config/modelpool/OpenCLIPVisionModelPool/ViT-B-16_TA8.yaml
fusion_bench/../fusion_bench_config/modelpool/OpenCLIPVisionModelPool/ViT-B-32_TA8.yaml
fusion_bench/../fusion_bench_config/modelpool/OpenCLIPVisionModelPool/ViT-B-32_TA_cars_dtd.yaml
fusion_bench/../fusion_bench_config/modelpool/OpenCLIPVisionModelPool/ViT-B-32_TA_sun397_cars.yaml
fusion_bench/../fusion_bench_config/modelpool/OpenCLIPVisionModelPool/ViT-B-32_TA_sun397_dtd.yaml
fusion_bench/../fusion_bench_config/modelpool/OpenCLIPVisionModelPool/ViT-B-32_individual.yaml
fusion_bench/../fusion_bench_config/modelpool/OpenCLIPVisionModelPool/ViT-L-14_TA8.yaml
fusion_bench/../fusion_bench_config/modelpool/Seq2SeqLMPool/_template.yaml
fusion_bench/../fusion_bench_config/modelpool/Seq2SeqLMPool/flan-t5-base_glue.yaml
fusion_bench/../fusion_bench_config/modelpool/Seq2SeqLMPool/flan-t5-base_glue_lora16.yaml
fusion_bench/../fusion_bench_config/modelpool/Seq2SeqLMPool/flan-t5-base_glue_lora16_tta.yaml
fusion_bench/../fusion_bench_config/modelpool/Seq2SeqLMPool/flan-t5-base_glue_tta.yaml
fusion_bench/../fusion_bench_config/modelpool/Seq2SeqLMPool/flan-t5-base_individual.yaml
fusion_bench/../fusion_bench_config/modelpool/Seq2SeqLMPool/flan-t5-large_glue_lora16.yaml
fusion_bench/../fusion_bench_config/modelpool/SequenceClassificationModelPool/llama_preference700k.yaml
fusion_bench/../fusion_bench_config/modelpool/SequenceClassificationModelPool/roberta-base_glue.yaml
fusion_bench/../fusion_bench_config/modelpool/SequenceClassificationModelPool/single_reward_model.yaml
fusion_bench/../fusion_bench_config/path/default.yaml
fusion_bench/../fusion_bench_config/taskpool/clip-vit-base-patch32_robustness_clean.yaml
fusion_bench/../fusion_bench_config/taskpool/clip-vit-base-patch32_robustness_corrupted.yaml
fusion_bench/../fusion_bench_config/taskpool/dummy.yaml
fusion_bench/../fusion_bench_config/taskpool/flan-t5_glue_text_generation.yaml
fusion_bench/../fusion_bench_config/taskpool/gpt-2_glue.yaml
fusion_bench/../fusion_bench_config/taskpool/nyuv2_taskpool.yaml
fusion_bench/../fusion_bench_config/taskpool/reward_model_evaluation.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/_template.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-base-patch32_robustness_corrupted.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-base-patch32_svhn_and_mnist.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-classification_TA8.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-classification_TA8_B16.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-classification_TA8_L14.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-classification_TA8_val.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-classification_TA8_with_control_task.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-classification_TALL14.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-classification_TALL20.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_cifar10.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_cifar100.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_dtd.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_emnist_letters.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_eurosat.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_fashion_mnist.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_fer2013.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_food101.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_gtsrb.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_kmnist.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_mnist.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_oxford-iiit-pet.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_oxford_flowers102.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_oxford_flowers102_val.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_pcam.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_rendered-sst2.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_resisc45.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_stanford-cars.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_stl10.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_sun397.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip-vit-single-task_svhn.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip_rankone_wemoe_clip-vit-classification_TA8.yaml
fusion_bench/../fusion_bench_config/taskpool/CLIPVisionModelTaskPool/clip_sparse_wemoe_clip-vit-classification_TA8.yaml
fusion_bench/../fusion_bench_config/taskpool/LMEvalHarnessTaskPool/lm_eval.yaml
fusion_bench/../fusion_bench_config/taskpool/OpenCLIPVisionModelTaskPool/ViT-B-16_TA8.yaml
fusion_bench/../fusion_bench_config/taskpool/OpenCLIPVisionModelTaskPool/ViT-B-32_TA8.yaml
fusion_bench/../fusion_bench_config/taskpool/OpenCLIPVisionModelTaskPool/ViT-L-14_TA8.yaml
fusion_bench/_get_started/__init__.py
fusion_bench/_get_started/greeting_program.py
fusion_bench/compat/__init__.py
fusion_bench/compat/method/__init__.py
fusion_bench/compat/method/base_algorithm.py
fusion_bench/compat/modelpool/AutoModelForSeq2SeqLM.py
fusion_bench/compat/modelpool/__init__.py
fusion_bench/compat/modelpool/base_pool.py
fusion_bench/compat/modelpool/huggingface_clip_vision.py
fusion_bench/compat/taskpool/__init__.py
fusion_bench/compat/taskpool/base_pool.py
fusion_bench/compat/taskpool/clip_image_classification.py
fusion_bench/compat/taskpool/flan_t5_glue_text_generation.py
fusion_bench/constants/__init__.py
fusion_bench/constants/banner.py
fusion_bench/constants/clip_vision.py
fusion_bench/constants/paths.py
fusion_bench/constants/runtime.py
fusion_bench/dataset/__init__.py
fusion_bench/dataset/clip_dataset.py
fusion_bench/dataset/fer2013.py
fusion_bench/dataset/gpt2_glue.py
fusion_bench/dataset/gsm8k.py
fusion_bench/dataset/image_dataset.py
fusion_bench/dataset/imdb.py
fusion_bench/dataset/nyuv2.py
fusion_bench/dataset/arc_agi/__init__.py
fusion_bench/dataset/arc_agi/arc.py
fusion_bench/dataset/arc_agi/arc_agi.py
fusion_bench/dataset/arc_agi/augmenters.py
fusion_bench/dataset/arc_agi/messagers.py
fusion_bench/dataset/arc_agi/np_cache.py
fusion_bench/dataset/arc_agi/preprocess.py
fusion_bench/dataset/arc_agi/representers.py
fusion_bench/dataset/image_corruption/__init__.py
fusion_bench/dataset/image_corruption/make_corruption.py
fusion_bench/dataset/llama/__init__.py
fusion_bench/dataset/llama/alpaca.py
fusion_bench/dataset/llama/collate.py
fusion_bench/dataset/llama/metamathqa.py
fusion_bench/dataset/llama/openai.py
fusion_bench/dataset/llama/preference_700k.py
fusion_bench/dataset/llama/sharegpt.py
fusion_bench/dataset/llama/squad.py
fusion_bench/dataset/llama/stanford_shp.py
fusion_bench/dataset/llama/ultrachat.py
fusion_bench/dataset/llama/wikitext.py
fusion_bench/dataset/llama/utils/__init__.py
fusion_bench/method/__init__.py
fusion_bench/method/base_algorithm.py
fusion_bench/method/dummy.py
fusion_bench/method/ensemble.py
fusion_bench/method/model_recombination.py
fusion_bench/method/simple_average.py
fusion_bench/method/ada_svd/__init__.py
fusion_bench/method/ada_svd/clip_vision.py
fusion_bench/method/adamerging/__init__.py
fusion_bench/method/adamerging/clip_layer_wise_adamerging.py
fusion_bench/method/adamerging/clip_task_wise_adamerging.py
fusion_bench/method/adamerging/entropy_loss.py
fusion_bench/method/adamerging/flan_t5_layer_wise_adamerging.py
fusion_bench/method/adamerging/gpt2_layer_wise_adamerging.py
fusion_bench/method/adamerging/layer_wise_adamerging.py
fusion_bench/method/adamerging/llama_adamerging.py
fusion_bench/method/adamerging/min_norm_solvers.py
fusion_bench/method/adamerging/task_wise_adamerging.py
fusion_bench/method/adamerging/utils.py
fusion_bench/method/analysis/__init__.py
fusion_bench/method/analysis/task_vector_cos_similarity.py
fusion_bench/method/analysis/task_vector_violin_plot.py
fusion_bench/method/bitdelta/__init__.py
fusion_bench/method/bitdelta/bitdelta.py
fusion_bench/method/bitdelta/bitdelta_utils/__init__.py
fusion_bench/method/bitdelta/bitdelta_utils/binary_gemm_kernel.py
fusion_bench/method/bitdelta/bitdelta_utils/data.py
fusion_bench/method/bitdelta/bitdelta_utils/diff.py
fusion_bench/method/classification/__init__.py
fusion_bench/method/classification/clip_finetune.py
fusion_bench/method/classification/continual_clip_finetune.py
fusion_bench/method/concrete_subspace/__init__.py
fusion_bench/method/concrete_subspace/clip_concrete_adamerging.py
fusion_bench/method/concrete_subspace/clip_concrete_task_arithmetic.py
fusion_bench/method/concrete_subspace/clip_post_defense.py
fusion_bench/method/concrete_subspace/clip_safe_concrete_adamerging.py
fusion_bench/method/dare/__init__.py
fusion_bench/method/dare/simple_average.py
fusion_bench/method/dare/task_arithmetic.py
fusion_bench/method/dare/ties_merging.py
fusion_bench/method/dare/utils.py
fusion_bench/method/dawe/__init__.py
fusion_bench/method/dawe/dawe_for_clip.py
fusion_bench/method/dawe/warppers/__init__.py
fusion_bench/method/dawe/warppers/dawe_model.py
fusion_bench/method/depth_upscaling/__init__.py
fusion_bench/method/depth_upscaling/depth_upscaling.py
fusion_bench/method/depth_upscaling/depth_upscaling_for_llama.py
fusion_bench/method/doge_ta/__init__.py
fusion_bench/method/doge_ta/clip_layer_wise_adamerging.py
fusion_bench/method/doge_ta/doge_ta.py
fusion_bench/method/doge_ta/layer_wise_adamerging.py
fusion_bench/method/expert_sparsity/__init__.py
fusion_bench/method/expert_sparsity/mixtral/__init__.py
fusion_bench/method/expert_sparsity/mixtral/dynamic_skipping.py
fusion_bench/method/expert_sparsity/mixtral/layer_wise_pruning.py
fusion_bench/method/expert_sparsity/mixtral/progressive_pruning.py
fusion_bench/method/expert_sparsity/utils/calibration_data.py
fusion_bench/method/fisher_merging/__init__.py
fusion_bench/method/fisher_merging/clip_fisher_merging.py
fusion_bench/method/fisher_merging/fisher_merging.py
fusion_bench/method/fisher_merging/gpt2_fisher_merging.py
fusion_bench/method/fw_merging/__init__.py
fusion_bench/method/fw_merging/fw_hard.py
fusion_bench/method/fw_merging/fw_soft.py
fusion_bench/method/fw_merging/utils.py
fusion_bench/method/gossip/__init__.py
fusion_bench/method/gossip/clip_layer_wise_gossip.py
fusion_bench/method/gossip/clip_task_wise_gossip.py
fusion_bench/method/gossip/entropy_loss.py
fusion_bench/method/gossip/flan_t5_layer_wise_gossip.py
fusion_bench/method/gossip/layer_wise_gossip.py
fusion_bench/method/gossip/min_norm_solvers.py
fusion_bench/method/gossip/task_wise_gossip.py
fusion_bench/method/gossip/utils.py
fusion_bench/method/isotropic_merging/__init__.py
fusion_bench/method/isotropic_merging/iso.py
fusion_bench/method/isotropic_merging/iso_utils.py
fusion_bench/method/knots/__init__.py
fusion_bench/method/knots/knots_utils.py
fusion_bench/method/linear/__init__.py
fusion_bench/method/linear/expo.py
fusion_bench/method/linear/linear_interpolation.py
fusion_bench/method/linear/llama_expo.py
fusion_bench/method/linear/simple_average_for_llama.py
fusion_bench/method/linear/task_arithmetic_for_llama.py
fusion_bench/method/lm_finetune/__init__.py
fusion_bench/method/lm_finetune/bradley_terry_rm.py
fusion_bench/method/lm_finetune/causal_lm_pretrain.py
fusion_bench/method/lm_finetune/fullfinetune_sft.py
fusion_bench/method/lm_finetune/peftfinetune_sft.py
fusion_bench/method/mixture_of_experts/__init__.py
fusion_bench/method/mixture_of_experts/mixtral_merging.py
fusion_bench/method/mixture_of_experts/mixtral_upcycling.py
fusion_bench/method/model_stock/__init__.py
fusion_bench/method/model_stock/model_stock.py
fusion_bench/method/moe_pruner/__init__.py
fusion_bench/method/moe_pruner/moe_pruner.py
fusion_bench/method/moe_pruner/hooks/__init__.py
fusion_bench/method/moe_pruner/hooks/deepseek_v2.py
fusion_bench/method/moe_pruner/hooks/hook.py
fusion_bench/method/moe_pruner/hooks/mixtral.py
fusion_bench/method/moe_pruner/utils/__init__.py
fusion_bench/method/moe_pruner/utils/data.py
fusion_bench/method/moe_pruner/utils/layerwrapper.py
fusion_bench/method/moe_pruner/utils/prune.py
fusion_bench/method/moe_pruner/utils/score.py
fusion_bench/method/opcm/__init__.py
fusion_bench/method/opcm/opcm.py
fusion_bench/method/opcm/task_arithmetic.py
fusion_bench/method/opcm/ties_merging.py
fusion_bench/method/opcm/utils.py
fusion_bench/method/opcm/weight_average.py
fusion_bench/method/pruning/__init__.py
fusion_bench/method/pruning/llama_magnitude_prune.py
fusion_bench/method/pruning/llama_random_prune.py
fusion_bench/method/pruning/llama_sparsegpt_prune.py
fusion_bench/method/pruning/llama_wanda_prune.py
fusion_bench/method/pruning/magnitude_diff_pruning.py
fusion_bench/method/pruning/prune_utils.py
fusion_bench/method/pruning/sparsegpt_utils/__init__.py
fusion_bench/method/pruning/sparsegpt_utils/sparsegpt.py
fusion_bench/method/pruning/wanda_utils/__init__.py
fusion_bench/method/pruning/wanda_utils/ablate.py
fusion_bench/method/pruning/wanda_utils/data.py
fusion_bench/method/pruning/wanda_utils/eval.py
fusion_bench/method/pruning/wanda_utils/layerwrapper.py
fusion_bench/method/pruning/wanda_utils/prune.py
fusion_bench/method/pruning/wanda_utils/prune_opt.py
fusion_bench/method/pruning/wanda_utils/sparsegpt.py
fusion_bench/method/pwe_moe/__init__.py
fusion_bench/method/pwe_moe/clip_pwe_moe.py
fusion_bench/method/pwe_moe/module.py
fusion_bench/method/pwe_moe/openclip_pwe_moe.py
fusion_bench/method/pwe_moe/utils.py
fusion_bench/method/pwe_moe/phn/__init__.py
fusion_bench/method/pwe_moe/phn/solvers.py
fusion_bench/method/randes/__init__.py
fusion_bench/method/randes/base_algorithm.py
fusion_bench/method/randes/modelsoup.py
fusion_bench/method/randes/task_arithmetic.py
fusion_bench/method/rankone_moe/__init__.py
fusion_bench/method/rankone_moe/clip_rankone_moe.py
fusion_bench/method/rankone_moe/rankone_moe.py
fusion_bench/method/regmean/__init__.py
fusion_bench/method/regmean/clip_regmean.py
fusion_bench/method/regmean/gpt2_regmean.py
fusion_bench/method/regmean/regmean.py
fusion_bench/method/regmean/utils.py
fusion_bench/method/regmean_plusplus/__init__.py
fusion_bench/method/regmean_plusplus/clip_regmean_plusplus.py
fusion_bench/method/regmean_plusplus/regmean_plusplus.py
fusion_bench/method/slerp/__init__.py
fusion_bench/method/slerp/slerp.py
fusion_bench/method/slerp/slerp_utils.py
fusion_bench/method/smile_upscaling/__init__.py
fusion_bench/method/smile_upscaling/causal_lm_upscaling.py
fusion_bench/method/smile_upscaling/error_accumulation.py
fusion_bench/method/smile_upscaling/projected_energy.py
fusion_bench/method/smile_upscaling/singular_projection_merging.py
fusion_bench/method/smile_upscaling/smile_mistral_upscaling.py
fusion_bench/method/smile_upscaling/smile_qwen2_upscaling.py
fusion_bench/method/smile_upscaling/smile_upscaling.py
fusion_bench/method/sparse_we_moe/__init__.py
fusion_bench/method/sparse_we_moe/sparse_clip_we_moe.py
fusion_bench/method/sparse_we_moe/sparse_we_moe.py
fusion_bench/method/sparselo/__init__.py
fusion_bench/method/sparselo/sparselo.py
fusion_bench/method/surgery/__init__.py
fusion_bench/method/surgery/clip_layer_wise_adamerging_surgery.py
fusion_bench/method/tall_mask/__init__.py
fusion_bench/method/tall_mask/task_arithmetic.py
fusion_bench/method/tall_mask/utils.py
fusion_bench/method/task_arithmetic/__init__.py
fusion_bench/method/task_arithmetic/task_arithmetic.py
fusion_bench/method/task_singular_vector/TSVC.py
fusion_bench/method/task_singular_vector/TSVM.py
fusion_bench/method/task_singular_vector/__init__.py
fusion_bench/method/task_singular_vector/utils/TSVC_utils.py
fusion_bench/method/task_singular_vector/utils/TSVM_utils.py
fusion_bench/method/task_singular_vector/utils/__init__.py
fusion_bench/method/task_singular_vector/utils/task_singular_interference.py
fusion_bench/method/ties_merging/__init__.py
fusion_bench/method/ties_merging/ties_merging.py
fusion_bench/method/ties_merging/ties_merging_utils.py
fusion_bench/method/trust_region/__init__.py
fusion_bench/method/trust_region/clip_task_arithmetic.py
fusion_bench/method/trust_region/utils.py
fusion_bench/method/we_moe/__init__.py
fusion_bench/method/we_moe/clip_we_moe.py
fusion_bench/method/we_moe/entropy_loss.py
fusion_bench/method/we_moe/flan_t5_we_moe.py
fusion_bench/method/we_moe/utils.py
fusion_bench/method/we_moe/we_moe.py
fusion_bench/method/weighted_average/__init__.py
fusion_bench/method/weighted_average/llama.py
fusion_bench/method/weighted_average/weighted_average.py
fusion_bench/metrics/__init__.py
fusion_bench/metrics/continual_learning/__init__.py
fusion_bench/metrics/continual_learning/backward_transfer.py
fusion_bench/metrics/nyuv2/__init__.py
fusion_bench/metrics/nyuv2/depth.py
fusion_bench/metrics/nyuv2/loss.py
fusion_bench/metrics/nyuv2/noise.py
fusion_bench/metrics/nyuv2/normal.py
fusion_bench/metrics/nyuv2/segmentation.py
fusion_bench/metrics/text_to_image_generation/__init__.py
fusion_bench/metrics/text_to_image_generation/aesthetic_scorer.py
fusion_bench/metrics/text_to_image_generation/compressibility.py
fusion_bench/metrics/text_to_image_generation/pickscore_scorer.py
fusion_bench/mixins/__init__.py
fusion_bench/mixins/clip_classification.py
fusion_bench/mixins/fabric_training.py
fusion_bench/mixins/hydra_config.py
fusion_bench/mixins/lightning_fabric.py
fusion_bench/mixins/openclip_classification.py
fusion_bench/mixins/rich_live.py
fusion_bench/mixins/serialization.py
fusion_bench/mixins/simple_profiler.py
fusion_bench/mixins/optim/__init__.py
fusion_bench/mixins/optim/adamw_with_warmup.py
fusion_bench/modelpool/PeftModelForSeq2SeqLM.py
fusion_bench/modelpool/__init__.py
fusion_bench/modelpool/base_pool.py
fusion_bench/modelpool/huggingface_automodel.py
fusion_bench/modelpool/huggingface_gpt2_classification.py
fusion_bench/modelpool/lazy_state_dict_pool.py
fusion_bench/modelpool/nyuv2_modelpool.py
fusion_bench/modelpool/causal_lm/__init__.py
fusion_bench/modelpool/causal_lm/causal_lm.py
fusion_bench/modelpool/clip_vision/__init__.py
fusion_bench/modelpool/clip_vision/modelpool.py
fusion_bench/modelpool/openclip_vision/__init__.py
fusion_bench/modelpool/openclip_vision/modelpool.py
fusion_bench/modelpool/seq2seq_lm/__init__.py
fusion_bench/modelpool/seq2seq_lm/modelpool.py
fusion_bench/modelpool/seq_classification_lm/__init__.py
fusion_bench/modelpool/seq_classification_lm/reward_model.py
fusion_bench/modelpool/seq_classification_lm/seq_classification_lm.py
fusion_bench/models/__init__.py
fusion_bench/models/hf_clip.py
fusion_bench/models/hf_utils.py
fusion_bench/models/parameter_dict.py
fusion_bench/models/rankone_moe.py
fusion_bench/models/separate_io.py
fusion_bench/models/sparse_we_moe.py
fusion_bench/models/utils.py
fusion_bench/models/we_moe.py
fusion_bench/models/chat_templates/__init__.py
fusion_bench/models/chat_templates/llama_3_Instruct.py
fusion_bench/models/chat_templates/load_tokenizer.py
fusion_bench/models/expert_sparsity/__init__.py
fusion_bench/models/expert_sparsity/mixtral/__init__.py
fusion_bench/models/expert_sparsity/mixtral/dataset.py
fusion_bench/models/expert_sparsity/mixtral/modeling_mixtral.py
fusion_bench/models/expert_sparsity/mixtral/wrapper.py
fusion_bench/models/linearized/__init__.py
fusion_bench/models/linearized/linearized_model_utils.py
fusion_bench/models/linearized/vision_model.py
fusion_bench/models/llama/__init__.py
fusion_bench/models/llama/patcher.py
fusion_bench/models/llama/tokenizer_loader.py
fusion_bench/models/llama/model_utils/__init__.py
fusion_bench/models/llama/model_utils/embedding.py
fusion_bench/models/llama/model_utils/liger_kernel.py
fusion_bench/models/llama/model_utils/misc.py
fusion_bench/models/llama/model_utils/mod.py
fusion_bench/models/llama/model_utils/visual.py
fusion_bench/models/masks/__init__.py
fusion_bench/models/masks/mask_model.py
fusion_bench/models/model_card_templates/default.md
fusion_bench/models/modeling_deepseek_v2/__init__.py
fusion_bench/models/modeling_deepseek_v2/configuration_deepseek.py
fusion_bench/models/modeling_deepseek_v2/modeling_deepseek.py
fusion_bench/models/modeling_deepseek_v2/tokenization_deepseek_fast.py
fusion_bench/models/modeling_losparse_llama/__init__.py
fusion_bench/models/modeling_losparse_llama/configuration_losparse_llama.py
fusion_bench/models/modeling_losparse_llama/losparse_linear.py
fusion_bench/models/modeling_losparse_llama/modeling_losparse_llama.py
fusion_bench/models/modeling_losparse_llama/register.py
fusion_bench/models/modeling_losparse_llama/utils.py
fusion_bench/models/modeling_smile_gemma2/__init__.py
fusion_bench/models/modeling_smile_gemma2/configuration_smile_gemma2.py
fusion_bench/models/modeling_smile_gemma2/modeling_smile_gemma2.py
fusion_bench/models/modeling_smile_gemma2/register.py
fusion_bench/models/modeling_smile_llama/__init__.py
fusion_bench/models/modeling_smile_llama/configuration_smile_llama.py
fusion_bench/models/modeling_smile_llama/modeling_smile_llama.py
fusion_bench/models/modeling_smile_llama/register.py
fusion_bench/models/modeling_smile_mistral/__init__.py
fusion_bench/models/modeling_smile_mistral/configuration_smile_mistral.py
fusion_bench/models/modeling_smile_mistral/modeling_smile_mistral.py
fusion_bench/models/modeling_smile_mistral/register.py
fusion_bench/models/modeling_smile_qwen2/__init__.py
fusion_bench/models/modeling_smile_qwen2/configuration_smile_qwen2.py
fusion_bench/models/modeling_smile_qwen2/modeling_smile_qwen2.py
fusion_bench/models/modeling_smile_qwen2/register.py
fusion_bench/models/nyuv2/__init__.py
fusion_bench/models/nyuv2/aspp.py
fusion_bench/models/nyuv2/lightning_module.py
fusion_bench/models/nyuv2/resnet.py
fusion_bench/models/nyuv2/resnet_dilated.py
fusion_bench/models/open_clip/__init__.py
fusion_bench/models/open_clip/modeling.py
fusion_bench/models/open_clip/utils.py
fusion_bench/models/open_clip/variables_and_paths.py
fusion_bench/models/smile_moe/__init__.py
fusion_bench/models/smile_moe/linear_from_hf_config.py
fusion_bench/models/smile_moe/linear_from_module.py
fusion_bench/models/smile_moe/utils/__init__.py
fusion_bench/models/smile_moe/utils/svd_utils.py
fusion_bench/models/surgery/__init__.py
fusion_bench/models/surgery/surgerymodelwrapper.py
fusion_bench/models/wrappers/__init__.py
fusion_bench/models/wrappers/ensemble.py
fusion_bench/models/wrappers/layer_wise_fusion.py
fusion_bench/models/wrappers/layer_wise_fusion_doge_ta.py
fusion_bench/models/wrappers/task_wise_fusion.py
fusion_bench/optim/__init__.py
fusion_bench/optim/exception.py
fusion_bench/optim/mezo.py
fusion_bench/optim/lr_scheduler/__init__.py
fusion_bench/optim/lr_scheduler/linear_warmup.py
fusion_bench/optim/lr_scheduler/utils/__init__.py
fusion_bench/optim/lr_scheduler/utils/visualization.py
fusion_bench/programs/__init__.py
fusion_bench/programs/base_program.py
fusion_bench/programs/fabric_fusion_program.py
fusion_bench/scripts/__init__.py
fusion_bench/scripts/cli.py
fusion_bench/scripts/imgui.py
fusion_bench/scripts/nyuv2_mtl_train.py
fusion_bench/scripts/webui.py
fusion_bench/scripts/clip/__init__.py
fusion_bench/scripts/clip/convert_checkpoint.py
fusion_bench/taskpool/__init__.py
fusion_bench/taskpool/base_pool.py
fusion_bench/taskpool/dummy.py
fusion_bench/taskpool/gpt2_text_classification.py
fusion_bench/taskpool/nyuv2_taskpool.py
fusion_bench/taskpool/clip_vision/__init__.py
fusion_bench/taskpool/clip_vision/clip_rankone_moe_taskpool.py
fusion_bench/taskpool/clip_vision/clip_smile_taskpool.py
fusion_bench/taskpool/clip_vision/clip_sparse_wemoe_taskpool.py
fusion_bench/taskpool/clip_vision/taskpool.py
fusion_bench/taskpool/clip_vision/utils/__init__.py
fusion_bench/taskpool/clip_vision/utils/routing_analysis_utils.py
fusion_bench/taskpool/llama/__init__.py
fusion_bench/taskpool/llama/reward_model.py
fusion_bench/taskpool/llama/test_generation.py
fusion_bench/taskpool/lm_eval_harness/__init__.py
fusion_bench/taskpool/lm_eval_harness/taskpool.py
fusion_bench/taskpool/openclip_vision/__init__.py
fusion_bench/taskpool/openclip_vision/openclip_taskpool.py
fusion_bench/tasks/__init__.py
fusion_bench/tasks/base_task.py
fusion_bench/tasks/classification.py
fusion_bench/tasks/clip_classification/__init__.py
fusion_bench/tasks/clip_classification/cifar10.py
fusion_bench/tasks/clip_classification/cifar100.py
fusion_bench/tasks/clip_classification/clip_dataset.py
fusion_bench/tasks/clip_classification/cub_200_2011.py
fusion_bench/tasks/clip_classification/dtd.py
fusion_bench/tasks/clip_classification/emnist_letters.py
fusion_bench/tasks/clip_classification/emnist_mnist.py
fusion_bench/tasks/clip_classification/eurosat.py
fusion_bench/tasks/clip_classification/fashion_mnist.py
fusion_bench/tasks/clip_classification/fer2013.py
fusion_bench/tasks/clip_classification/flower102.py
fusion_bench/tasks/clip_classification/food101.py
fusion_bench/tasks/clip_classification/gtsrb.py
fusion_bench/tasks/clip_classification/imagenet.py
fusion_bench/tasks/clip_classification/kmnist.py
fusion_bench/tasks/clip_classification/mnist.py
fusion_bench/tasks/clip_classification/mongo_leaf_disease.py
fusion_bench/tasks/clip_classification/oxford_iiit_pet.py
fusion_bench/tasks/clip_classification/pcam.py
fusion_bench/tasks/clip_classification/rendered_sst2.py
fusion_bench/tasks/clip_classification/resisc45.py
fusion_bench/tasks/clip_classification/stanford_cars.py
fusion_bench/tasks/clip_classification/stl10.py
fusion_bench/tasks/clip_classification/sun397.py
fusion_bench/tasks/clip_classification/svhn.py
fusion_bench/tasks/clip_classification/tiny_imagenet.py
fusion_bench/tasks/flan_t5_text_generation/__init__.py
fusion_bench/tasks/flan_t5_text_generation/datasets_preprocess.py
fusion_bench/tasks/flan_t5_text_generation/glue_evaluation.py
fusion_bench/tasks/flan_t5_text_generation/glue_load_dataset.py
fusion_bench/tasks/flan_t5_text_generation/glue_preprocessors.py
fusion_bench/tasks/flan_t5_text_generation/glue_prompt_templates.py
fusion_bench/utils/__init__.py
fusion_bench/utils/auto.py
fusion_bench/utils/cache_utils.py
fusion_bench/utils/data.py
fusion_bench/utils/devices.py
fusion_bench/utils/dict.py
fusion_bench/utils/dtype.py
fusion_bench/utils/expr.py
fusion_bench/utils/fabric.py
fusion_bench/utils/functools.py
fusion_bench/utils/hydra_utils.py
fusion_bench/utils/instantiate_utils.py
fusion_bench/utils/json.py
fusion_bench/utils/lazy_imports.py
fusion_bench/utils/lazy_state_dict.py
fusion_bench/utils/misc.py
fusion_bench/utils/modelscope.py
fusion_bench/utils/packages.py
fusion_bench/utils/parameters.py
fusion_bench/utils/path.py
fusion_bench/utils/pylogger.py
fusion_bench/utils/rich_utils.py
fusion_bench/utils/set.py
fusion_bench/utils/state_dict_arithmetic.py
fusion_bench/utils/tensorboard.py
fusion_bench/utils/timer.py
fusion_bench/utils/type.py
fusion_bench/utils/plot/__init__.py
fusion_bench/utils/plot/color_data.py
fusion_bench/utils/plot/token.py
fusion_bench/utils/plot/token_notebook.py
fusion_bench/utils/strenum/__init__.py
fusion_bench/utils/strenum/_name_mangler.py
fusion_bench/utils/strenum/_version.py
tests/test_depth_upscaling.py
tests/test_simple_average.py
tests/test_weighed_ensemble.py