Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-24 20:19:22 +01:00
Code Issues Packages Projects Releases Wiki Activity
Files
de33f4dc874f17cabde4e144381e71fb3b25426e
modelscope/tests/trainers
History
Wang Qiang de33f4dc87 Support sdxl finetune by lora method (#468) 
* support sdxl finetune by lora

* remove useless imports

* support sdxl finetune

* upgrade diffusers to 0.19.0

* sdxl finetune

* fix bugs

* pre commit

* diffusers>=0.19.0
2023-08-23 11:23:34 +08:00
..
audio
[to #42322933] add DFSMN model to far field KWS
2023-04-20 22:45:07 +08:00
cli
Fix CI: test merge dataset failed
2023-05-29 10:32:33 +08:00
hooks
Support FlexTrain and update the structure of trainer
2023-05-13 12:12:04 +08:00
lrscheduler
[to #43112534] finetune support and first case
2022-07-14 16:25:55 +08:00
utils
[to #42322933] Refactor text generation model outputs and fix some bugs
2022-10-27 09:52:05 +08:00
__init__.py
[to #41401401] modelhub and Trainer support
2022-05-30 11:53:53 +08:00
model_trainer_map.py
Add trainer for UniTE
2023-05-11 14:41:08 +08:00
test_action_detection_trainer.py
fix bug for install detecron2
2023-03-10 15:14:21 +08:00
test_card_detection_scrfd_trainer.py
[to #42322933]更新face_detection_scrfd模型并支持finetune, 新增card_detection模型
2022-10-12 19:58:50 +08:00
test_clip_trainer.py
[to #42322933]clip支持finetune
2022-10-30 21:51:11 +08:00
test_custom_diffusion_trainer.py
fix bugs
2023-07-24 19:46:22 +08:00
test_dialog_intent_trainer.py
skip tests/trainers/test_dialog_intent_trainer.py for list model file 500 error
2023-01-06 09:15:50 +08:00
test_dialog_modeling_trainer.py
Add teardown for tests
2023-06-28 09:44:44 +08:00
test_document_grounded_dialog_generate_trainer.py
[to #42322933] Doc2Bot documentation with retrieval rerank, generation
2023-02-12 03:09:33 +00:00
test_document_grounded_dialog_rerank_trainer.py
Add teardown for tests
2023-06-28 09:44:44 +08:00
test_document_grounded_dialog_retrieval_trainer.py
[to #42322933] Doc2Bot documentation with retrieval rerank, generation
2023-02-12 03:09:33 +00:00
test_dreambooth_diffusion_trainer.py
Dreambooth method for finetuning stable diffusions (#339)
2023-06-28 20:10:28 +08:00
test_efficient_diffusion_tuning_trainer_swift.py
Switching model from huggingface to modelscope hub of efficient tuning (#479)
2023-08-18 20:30:04 +08:00
test_efficient_diffusion_tuning_trainer.py
Switching model from huggingface to modelscope hub of efficient tuning (#479)
2023-08-18 20:30:04 +08:00
test_face_detection_damofd_trainer.py
[to #42322933] add damofd model
2023-04-11 11:08:59 +08:00
test_face_detection_scrfd_trainer.py
[to #42322933] remove revision usage for face detection
2022-10-25 09:28:01 +08:00
test_finetune_faq_question_answering.py
refactor faq model and add MGIMN model
2023-02-09 08:29:19 +00:00
test_finetune_gpt3_smoke_test.py
add first case for gpt3 test (#236)
2023-04-09 21:46:48 +08:00
test_finetune_gpt3.py
[to #42322933] Support multi-machine data and tensor parallel finetuning
2023-02-28 18:47:17 +08:00
test_finetune_gpt_moe.py
add gpt-moe model for modelscope finetune
2022-12-17 05:52:57 +08:00
test_finetune_mgeo.py
fix failed case
2023-02-10 10:14:24 +08:00
test_finetune_mplug.py
[to #42322933] Add mplug pretrained model
2022-12-04 15:27:50 +08:00
test_finetune_plug_mental.py
add plug mental model
2023-02-06 10:57:20 +00:00
test_finetune_sentence_embedding.py
sentence-embedding support finetune
2023-02-10 06:07:38 +00:00
test_finetune_sequence_classification.py
Add Lora/Adapter/Prompt and support for chatglm6B and chatglm2-6B
2023-06-27 14:38:18 +08:00
test_finetune_text_generation.py
[to #42322933] Fix evaluation oom
2023-01-12 13:02:54 +08:00
test_finetune_text_ranking.py
[to #42322933]修正finetune text ranking bugs
2022-10-26 09:44:25 +08:00
test_finetune_token_classification.py
Merge branch 'master-gitlab' into merge_master_internal_1207
2022-12-07 19:59:07 +08:00
test_finetune_vision_efficient_tuning_swift.py
Add Lora/Adapter/Prompt and support for chatglm6B and chatglm2-6B
2023-06-27 14:38:18 +08:00
test_finetune_vision_efficient_tuning.py
support vision efficient tuning finetune
2023-03-08 16:42:23 +08:00
test_general_image_classification_trainer.py
add ConvNeXt model
2023-01-09 06:56:05 +08:00
test_gpt3_1_layer_base_qa_cw_tp_1_2.py
add token classification example and gpt3 one layer test (#268)
2023-04-13 10:21:00 +08:00
test_gpt3_text-generation_chinese-base_ddp.py
add token classification example and gpt3 one layer test (#268)
2023-04-13 10:21:00 +08:00
test_gpt3_text-generation_chinese-base_train.py
add token classification example and gpt3 one layer test (#268)
2023-04-13 10:21:00 +08:00
test_image_color_enhance_trainer.py
[to #44340132] fix: ci case run out of gpu memory
2022-08-30 17:59:15 +08:00
test_image_colorization_trainer.py
update image colorization metric
2023-03-28 15:14:10 +08:00
test_image_deblur_trainer.py
Refactor the task_datasets module
2023-03-10 09:03:32 +08:00
test_image_defrcn_fewshot_trainer.py
refine cv_image_defrcn trainer to avoid failed
2023-02-10 07:10:59 +00:00
test_image_denoise_trainer.py
Refactor the task_datasets module
2023-03-10 09:03:32 +08:00
test_image_inpainting_trainer.py
[to #42322933]suport image inpainting
2022-10-11 20:49:13 +08:00
test_image_instance_segmentation_trainer.py
Refactor the task_datasets module
2023-03-10 09:03:32 +08:00
test_image_portrait_enhancement_trainer.py
Refactor the task_datasets module
2023-03-10 09:03:32 +08:00
test_image_portrait_stylization_trainer.py
Add teardown for tests
2023-06-28 09:44:44 +08:00
test_language_guided_video_summarization_trainer.py
Refactor the task_datasets module
2023-03-10 09:03:32 +08:00
test_lora_diffusion_trainer.py
Update diffusers version to 0.18.0 (#377)
2023-07-14 19:02:52 +08:00
test_lora_diffusion_xl_trainer.py
Support sdxl finetune by lora method (#468)
2023-08-23 11:23:34 +08:00
test_movie_scene_segmentation_trainer.py
fix memory leak bug in eval
2022-12-25 08:40:19 +08:00
test_nerf_recon_acc_trainer.py
Add teardown for tests
2023-06-28 09:44:44 +08:00
test_ocr_detection_db_trainer.py
Add teardown for tests
2023-06-28 09:44:44 +08:00
test_ocr_recognition_trainer.py
add ocr_reco_crnntrain
2023-03-09 21:14:09 +08:00
test_ofa_mmspeech_trainer.py
Add teardown for tests
2023-06-28 09:44:44 +08:00
test_ofa_trainer.py
Add teardown for tests
2023-06-28 09:44:44 +08:00
test_plug_finetune_text_generation.py
Add teardown for tests
2023-06-28 09:44:44 +08:00
test_referring_video_object_segmentation_trainer.py
Add teardown for tests
2023-06-28 09:44:44 +08:00
test_siamese_uie_trainer.py
Refactor the task_datasets module
2023-03-10 09:03:32 +08:00
test_stable_diffusion_trainer.py
Merge code from github
2023-06-21 10:22:06 +08:00
test_table_question_answering_trainer.py
add space-t trainer
2023-01-04 09:46:37 +08:00
test_team_transfer_trainer.py
Add teardown for tests
2023-06-28 09:44:44 +08:00
test_tinynas_damoyolo_trainer.py
Add teardown for tests
2023-06-28 09:44:44 +08:00
test_trainer_gpu.py
trainer support parallel_groups
2023-03-09 21:33:35 +08:00
test_trainer_with_nlp.py
Support FlexTrain and update the structure of trainer
2023-05-13 12:12:04 +08:00
test_trainer.py
Convert cfg during training
2023-03-09 22:27:44 +08:00
test_training_args.py
Support FlexTrain and update the structure of trainer
2023-05-13 12:12:04 +08:00
test_translation_evaluation_trainer.py
Add teardown for tests
2023-06-28 09:44:44 +08:00
test_translation_trainer.py
[to #42322933] add en-zh en-es es-en base translation models
2022-11-29 13:44:06 +08:00
test_video_summarization_trainer.py
Refactor the task_datasets module
2023-03-10 09:03:32 +08:00