LICENSE
MANIFEST.in
README.md
pyproject.toml
requirements.txt
setup.cfg
setup.py
unitorch.png
.pytest_cache/README.md
benchmarks/README.md
docs/icon.png
docs/overview.png
docs/assets/images/favicon.png
docs/search/search_index.json
examples/README.md
examples/configs/caption/blip.ini
examples/configs/caption/minigpt4.ini
examples/configs/caption/minigpt4_ds.ini
examples/configs/classification/bert.ini
examples/configs/classification/clip.ini
examples/configs/classification/image_clip.ini
examples/configs/classification/roberta.ini
examples/configs/classification/swin.ini
examples/configs/classification/text_clip.ini
examples/configs/classification/lora/bloom.ini
examples/configs/classification/lora/bloom_ds.ini
examples/configs/classification/lora/llama.ini
examples/configs/classification/lora/llama_ds.ini
examples/configs/deepspeed/adamw.fp32.json
examples/configs/deepspeed/adamw.json
examples/configs/detections/detr.ini
examples/configs/diffusion/image2image/stable-v1.5-controlnet-canny.ini
examples/configs/diffusion/image2image/stable-v1.5.ini
examples/configs/diffusion/image2image/stable-xl-base.ini
examples/configs/diffusion/inpainting/stable-v1.5-controlnet.ini
examples/configs/diffusion/inpainting/stable-v1.5.ini
examples/configs/diffusion/inpainting/stable-xl-base.ini
examples/configs/diffusion/resolution/stable-v1.5.ini
examples/configs/diffusion/text2image/blip2.ini
examples/configs/diffusion/text2image/dreambooth.ini
examples/configs/diffusion/text2image/stable-v1.5-controlnet-canny.ini
examples/configs/diffusion/text2image/stable-v1.5.ini
examples/configs/diffusion/text2image/stable-v2.1.ini
examples/configs/diffusion/text2image/stable-v2.ini
examples/configs/diffusion/text2image/stable-xl-base-refiner.ini
examples/configs/diffusion/text2image/stable-xl-base_ds.ini
examples/configs/diffusion/text2video/stable-v1.5-realistic-animate-v1.5.ini
examples/configs/generation/bart.ini
examples/configs/generation/bloom.gptq.ini
examples/configs/generation/bloom.ini
examples/configs/generation/llama.ini
examples/configs/generation/llama.quant.ini
examples/configs/generation/mbart.ini
examples/configs/generation/mt5.ini
examples/configs/generation/pegasus.ini
examples/configs/generation/t5.ini
examples/configs/generation/xpegasus.ini
examples/configs/generation/lora/bloom.ini
examples/configs/generation/lora/bloom_ds.ini
examples/configs/generation/lora/llama.ini
examples/configs/generation/lora/llama.quant.ini
examples/configs/generation/lora/llama_ds.ini
examples/configs/generation/lora/llama_ds.quant.ini
examples/configs/pretrain/clip.ini
examples/configs/quantization/4bit.json
examples/configs/quantization/8bit.json
examples/configs/segmentation/bria.ini
examples/configs/segmentation/sam.ini
examples/configs/services/zip_image/config.ini
examples/configs/services/zip_image/config_v2.ini
examples/fastapis/stable.ini
examples/webuis/all.ini
examples/webuis/animate.ini
examples/webuis/blip2.ini
examples/webuis/controlnet.ini
examples/webuis/controlnet_xl.ini
examples/webuis/llama.ini
examples/webuis/minigpt4.ini
examples/webuis/sam.ini
examples/webuis/stable.ini
examples/webuis/stable_xl.ini
examples/webuis/stable_xl_refiner.ini
notebooks/README.md
src/unitorch/__init__.py
src/unitorch.egg-info/PKG-INFO
src/unitorch.egg-info/SOURCES.txt
src/unitorch.egg-info/dependency_links.txt
src/unitorch.egg-info/entry_points.txt
src/unitorch.egg-info/requires.txt
src/unitorch.egg-info/top_level.txt
src/unitorch/cli/__init__.py
src/unitorch/cli/core.py
src/unitorch/cli/decorators.py
src/unitorch/cli/assets/icon.png
src/unitorch/cli/consoles/__init__.py
src/unitorch/cli/consoles/eval.py
src/unitorch/cli/consoles/fastapi.py
src/unitorch/cli/consoles/infer.py
src/unitorch/cli/consoles/launch.py
src/unitorch/cli/consoles/service.py
src/unitorch/cli/consoles/train.py
src/unitorch/cli/consoles/webui.py
src/unitorch/cli/datasets/__init__.py
src/unitorch/cli/datasets/hf.py
src/unitorch/cli/fastapis/__init__.py
src/unitorch/cli/fastapis/animate.py
src/unitorch/cli/fastapis/controlnet.py
src/unitorch/cli/fastapis/controlnet_xl.py
src/unitorch/cli/fastapis/sam.py
src/unitorch/cli/fastapis/stable.py
src/unitorch/cli/fastapis/stable_xl.py
src/unitorch/cli/fastapis/stable_xl_refiner.py
src/unitorch/cli/losses/__init__.py
src/unitorch/cli/losses/ranking.py
src/unitorch/cli/models/__init__.py
src/unitorch/cli/models/classification_utils.py
src/unitorch/cli/models/detection_utils.py
src/unitorch/cli/models/diffusion_utils.py
src/unitorch/cli/models/generation_utils.py
src/unitorch/cli/models/image_utils.py
src/unitorch/cli/models/label_utils.py
src/unitorch/cli/models/modeling_utils.py
src/unitorch/cli/models/processing_utils.py
src/unitorch/cli/models/random_utils.py
src/unitorch/cli/models/ranking_utils.py
src/unitorch/cli/models/segmentation_utils.py
src/unitorch/cli/models/bart/__init__.py
src/unitorch/cli/models/bart/modeling.py
src/unitorch/cli/models/bart/processing.py
src/unitorch/cli/models/beit/__init__.py
src/unitorch/cli/models/beit/modeling.py
src/unitorch/cli/models/beit/processing.py
src/unitorch/cli/models/bert/__init__.py
src/unitorch/cli/models/bert/modeling.py
src/unitorch/cli/models/bert/processing.py
src/unitorch/cli/models/blip/__init__.py
src/unitorch/cli/models/blip/modeling.py
src/unitorch/cli/models/blip/processing.py
src/unitorch/cli/models/bloom/__init__.py
src/unitorch/cli/models/bloom/modeling.py
src/unitorch/cli/models/bloom/processing.py
src/unitorch/cli/models/bria/__init__.py
src/unitorch/cli/models/bria/modeling.py
src/unitorch/cli/models/bria/processing.py
src/unitorch/cli/models/clip/__init__.py
src/unitorch/cli/models/clip/modeling.py
src/unitorch/cli/models/clip/processing.py
src/unitorch/cli/models/deberta/__init__.py
src/unitorch/cli/models/deberta/modeling.py
src/unitorch/cli/models/deberta/modeling_v2.py
src/unitorch/cli/models/deberta/processing.py
src/unitorch/cli/models/deberta/processing_v2.py
src/unitorch/cli/models/detr/__init__.py
src/unitorch/cli/models/detr/modeling.py
src/unitorch/cli/models/detr/processing.py
src/unitorch/cli/models/diffusers/__init__.py
src/unitorch/cli/models/diffusers/convert_checkpoint.py
src/unitorch/cli/models/diffusers/modeling_animate.py
src/unitorch/cli/models/diffusers/modeling_blip2.py
src/unitorch/cli/models/diffusers/modeling_controlnet.py
src/unitorch/cli/models/diffusers/modeling_controlnet_xl.py
src/unitorch/cli/models/diffusers/modeling_dreambooth.py
src/unitorch/cli/models/diffusers/modeling_dreambooth_xl.py
src/unitorch/cli/models/diffusers/modeling_stable.py
src/unitorch/cli/models/diffusers/modeling_stable_xl.py
src/unitorch/cli/models/diffusers/modeling_stable_xl_refiner.py
src/unitorch/cli/models/diffusers/processing_blip2.py
src/unitorch/cli/models/diffusers/processing_controlnet.py
src/unitorch/cli/models/diffusers/processing_controlnet_xl.py
src/unitorch/cli/models/diffusers/processing_dreambooth.py
src/unitorch/cli/models/diffusers/processing_dreambooth_xl.py
src/unitorch/cli/models/diffusers/processing_stable.py
src/unitorch/cli/models/diffusers/processing_stable_xl.py
src/unitorch/cli/models/diffusers/processing_stable_xl_refiner.py
src/unitorch/cli/models/llama/__init__.py
src/unitorch/cli/models/llama/modeling.py
src/unitorch/cli/models/llama/processing.py
src/unitorch/cli/models/mbart/__init__.py
src/unitorch/cli/models/mbart/modeling.py
src/unitorch/cli/models/mbart/processing.py
src/unitorch/cli/models/minigpt4/__init__.py
src/unitorch/cli/models/minigpt4/modeling.py
src/unitorch/cli/models/minigpt4/processing.py
src/unitorch/cli/models/mistral/__init__.py
src/unitorch/cli/models/mistral/modeling.py
src/unitorch/cli/models/mistral/processing.py
src/unitorch/cli/models/mt5/__init__.py
src/unitorch/cli/models/mt5/modeling.py
src/unitorch/cli/models/mt5/processing.py
src/unitorch/cli/models/peft/__init__.py
src/unitorch/cli/models/peft/modeling_bloom.py
src/unitorch/cli/models/peft/modeling_llama.py
src/unitorch/cli/models/peft/modeling_mistral.py
src/unitorch/cli/models/pegasus/__init__.py
src/unitorch/cli/models/pegasus/modeling.py
src/unitorch/cli/models/pegasus/processing.py
src/unitorch/cli/models/roberta/__init__.py
src/unitorch/cli/models/roberta/modeling.py
src/unitorch/cli/models/roberta/processing.py
src/unitorch/cli/models/sam/__init__.py
src/unitorch/cli/models/sam/modeling.py
src/unitorch/cli/models/sam/processing.py
src/unitorch/cli/models/swin/__init__.py
src/unitorch/cli/models/swin/modeling.py
src/unitorch/cli/models/swin/processing.py
src/unitorch/cli/models/t5/__init__.py
src/unitorch/cli/models/t5/modeling.py
src/unitorch/cli/models/t5/processing.py
src/unitorch/cli/models/visualbert/__init__.py
src/unitorch/cli/models/visualbert/modeling.py
src/unitorch/cli/models/visualbert/processing.py
src/unitorch/cli/models/vit/__init__.py
src/unitorch/cli/models/vit/modeling.py
src/unitorch/cli/models/vit/processing.py
src/unitorch/cli/models/xlm_roberta/__init__.py
src/unitorch/cli/models/xlm_roberta/modeling.py
src/unitorch/cli/models/xlm_roberta/processing.py
src/unitorch/cli/models/xpegasus/__init__.py
src/unitorch/cli/models/xpegasus/modeling.py
src/unitorch/cli/models/xpegasus/processing.py
src/unitorch/cli/optims/__init__.py
src/unitorch/cli/optims/lion.py
src/unitorch/cli/pipelines/__init__.py
src/unitorch/cli/pipelines/bloom.py
src/unitorch/cli/pipelines/llama.py
src/unitorch/cli/pipelines/minigpt4.py
src/unitorch/cli/pipelines/mistral.py
src/unitorch/cli/pipelines/sam.py
src/unitorch/cli/pipelines/animate/__init__.py
src/unitorch/cli/pipelines/animate/text2video.py
src/unitorch/cli/pipelines/blip2/__init__.py
src/unitorch/cli/pipelines/blip2/text2image.py
src/unitorch/cli/pipelines/blip2controlnet/__init__.py
src/unitorch/cli/pipelines/blip2controlnet/text2image.py
src/unitorch/cli/pipelines/controlnet/__init__.py
src/unitorch/cli/pipelines/controlnet/image2image.py
src/unitorch/cli/pipelines/controlnet/inpainting.py
src/unitorch/cli/pipelines/controlnet/text2image.py
src/unitorch/cli/pipelines/controlnet_xl/__init__.py
src/unitorch/cli/pipelines/controlnet_xl/image2image.py
src/unitorch/cli/pipelines/controlnet_xl/inpainting.py
src/unitorch/cli/pipelines/controlnet_xl/text2image.py
src/unitorch/cli/pipelines/peft/__init__.py
src/unitorch/cli/pipelines/peft/bloom.py
src/unitorch/cli/pipelines/peft/mistral.py
src/unitorch/cli/pipelines/stable/__init__.py
src/unitorch/cli/pipelines/stable/image2image.py
src/unitorch/cli/pipelines/stable/inpainting.py
src/unitorch/cli/pipelines/stable/resolution.py
src/unitorch/cli/pipelines/stable/text2image.py
src/unitorch/cli/pipelines/stable_xl/__init__.py
src/unitorch/cli/pipelines/stable_xl/image2image.py
src/unitorch/cli/pipelines/stable_xl/inpainting.py
src/unitorch/cli/pipelines/stable_xl/text2image.py
src/unitorch/cli/pipelines/stable_xl_refiner/__init__.py
src/unitorch/cli/pipelines/stable_xl_refiner/image2image.py
src/unitorch/cli/pipelines/stable_xl_refiner/inpainting.py
src/unitorch/cli/pipelines/stable_xl_refiner/text2image.py
src/unitorch/cli/schedulers/__init__.py
src/unitorch/cli/schedulers/warmup.py
src/unitorch/cli/scores/__init__.py
src/unitorch/cli/scripts/README.md
src/unitorch/cli/scripts/__init__.py
src/unitorch/cli/services/__init__.py
src/unitorch/cli/services/readme.md
src/unitorch/cli/services/zip_image/__init__.py
src/unitorch/cli/services/zip_image/service.py
src/unitorch/cli/tasks/__init__.py
src/unitorch/cli/tasks/deepspeed.py
src/unitorch/cli/tasks/supervised.py
src/unitorch/cli/utils/__init__.py
src/unitorch/cli/webuis/__init__.py
src/unitorch/cli/webuis/animate.py
src/unitorch/cli/webuis/blip2.py
src/unitorch/cli/webuis/bloom.py
src/unitorch/cli/webuis/controlnet.py
src/unitorch/cli/webuis/controlnet_xl.py
src/unitorch/cli/webuis/image_utils.py
src/unitorch/cli/webuis/llama.py
src/unitorch/cli/webuis/minigpt4.py
src/unitorch/cli/webuis/mistral.py
src/unitorch/cli/webuis/sam.py
src/unitorch/cli/webuis/stable.py
src/unitorch/cli/webuis/stable_xl.py
src/unitorch/cli/webuis/stable_xl_refiner.py
src/unitorch/cli/writers/__init__.py
src/unitorch/datasets/__init__.py
src/unitorch/datasets/hf.py
src/unitorch/losses/__init__.py
src/unitorch/losses/ranking.py
src/unitorch/models/__init__.py
src/unitorch/models/modeling_ema.py
src/unitorch/models/processing_utils.py
src/unitorch/models/quantization.py
src/unitorch/models/bart/__init__.py
src/unitorch/models/bart/modeling.py
src/unitorch/models/bart/processing.py
src/unitorch/models/beit/__init__.py
src/unitorch/models/beit/modeling.py
src/unitorch/models/beit/processing.py
src/unitorch/models/bert/__init__.py
src/unitorch/models/bert/modeling.py
src/unitorch/models/bert/processing.py
src/unitorch/models/blip/__init__.py
src/unitorch/models/blip/modeling.py
src/unitorch/models/blip/processing.py
src/unitorch/models/blip2/__init__.py
src/unitorch/models/blip2/modeling.py
src/unitorch/models/blip2/processing.py
src/unitorch/models/bloom/__init__.py
src/unitorch/models/bloom/modeling.py
src/unitorch/models/bloom/processing.py
src/unitorch/models/bria/__init__.py
src/unitorch/models/bria/modeling.py
src/unitorch/models/bria/processing.py
src/unitorch/models/clip/__init__.py
src/unitorch/models/clip/modeling.py
src/unitorch/models/clip/processing.py
src/unitorch/models/deberta/__init__.py
src/unitorch/models/deberta/modeling.py
src/unitorch/models/deberta/modeling_v2.py
src/unitorch/models/deberta/processing.py
src/unitorch/models/deberta/processing_v2.py
src/unitorch/models/detr/__init__.py
src/unitorch/models/detr/modeling.py
src/unitorch/models/detr/processing.py
src/unitorch/models/diffusers/__init__.py
src/unitorch/models/diffusers/modeling_animate.py
src/unitorch/models/diffusers/modeling_blip2.py
src/unitorch/models/diffusers/modeling_controlnet.py
src/unitorch/models/diffusers/modeling_controlnet_xl.py
src/unitorch/models/diffusers/modeling_dreambooth.py
src/unitorch/models/diffusers/modeling_dreambooth_xl.py
src/unitorch/models/diffusers/modeling_multicontrolnet.py
src/unitorch/models/diffusers/modeling_stable.py
src/unitorch/models/diffusers/modeling_stable_xl.py
src/unitorch/models/diffusers/modeling_stable_xl_refiner.py
src/unitorch/models/diffusers/processing_blip2.py
src/unitorch/models/diffusers/processing_controlnet.py
src/unitorch/models/diffusers/processing_controlnet_xl.py
src/unitorch/models/diffusers/processing_dreambooth.py
src/unitorch/models/diffusers/processing_dreambooth_xl.py
src/unitorch/models/diffusers/processing_multicontrolnet.py
src/unitorch/models/diffusers/processing_stable.py
src/unitorch/models/diffusers/processing_stable_xl.py
src/unitorch/models/diffusers/processing_stable_xl_refiner.py
src/unitorch/models/llama/__init__.py
src/unitorch/models/llama/modeling.py
src/unitorch/models/llama/processing.py
src/unitorch/models/mbart/__init__.py
src/unitorch/models/mbart/modeling.py
src/unitorch/models/mbart/processing.py
src/unitorch/models/minigpt4/__init__.py
src/unitorch/models/minigpt4/modeling.py
src/unitorch/models/minigpt4/processing.py
src/unitorch/models/mistral/__init__.py
src/unitorch/models/mistral/modeling.py
src/unitorch/models/mistral/processing.py
src/unitorch/models/mt5/__init__.py
src/unitorch/models/mt5/modeling.py
src/unitorch/models/mt5/processing.py
src/unitorch/models/peft/__init__.py
src/unitorch/models/peft/modeling_bloom.py
src/unitorch/models/peft/modeling_llama.py
src/unitorch/models/peft/modeling_mistral.py
src/unitorch/models/peft/processing.py
src/unitorch/models/pegasus/__init__.py
src/unitorch/models/pegasus/modeling.py
src/unitorch/models/pegasus/processing.py
src/unitorch/models/roberta/__init__.py
src/unitorch/models/roberta/modeling.py
src/unitorch/models/roberta/processing.py
src/unitorch/models/sam/__init__.py
src/unitorch/models/sam/modeling.py
src/unitorch/models/sam/processing.py
src/unitorch/models/swin/__init__.py
src/unitorch/models/swin/modeling.py
src/unitorch/models/swin/processing.py
src/unitorch/models/t5/__init__.py
src/unitorch/models/t5/modeling.py
src/unitorch/models/t5/processing.py
src/unitorch/models/visualbert/__init__.py
src/unitorch/models/visualbert/modeling.py
src/unitorch/models/visualbert/processing.py
src/unitorch/models/vit/__init__.py
src/unitorch/models/vit/modeling.py
src/unitorch/models/vit/processing.py
src/unitorch/models/xlm_roberta/__init__.py
src/unitorch/models/xlm_roberta/modeling.py
src/unitorch/models/xlm_roberta/processing.py
src/unitorch/models/xpegasus/__init__.py
src/unitorch/models/xpegasus/modeling.py
src/unitorch/models/xpegasus/processing.py
src/unitorch/modules/__init__.py
src/unitorch/modules/classifier.py
src/unitorch/modules/timm.py
src/unitorch/modules/replace/__init__.py
src/unitorch/modules/replace/datasets_v2.py
src/unitorch/modules/replace/hf_hub_v2.py
src/unitorch/ops/__init__.py
src/unitorch/ops/ngram_repeat_block.py
src/unitorch/optims/__init__.py
src/unitorch/optims/lion.py
src/unitorch/schedulers/__init__.py
src/unitorch/schedulers/warmup.py
src/unitorch/scores/__init__.py
src/unitorch/scores/bleu.py
src/unitorch/scores/map.py
src/unitorch/scores/rouge.py
src/unitorch/tasks/__init__.py
src/unitorch/utils/__init__.py
src/unitorch/utils/decorators.py
src/unitorch/utils/functional.py
src/unitorch/utils/image_utils.py
src/unitorch/utils/import_utils.py
src/unitorch/utils/io.py
src/unitorch/utils/palette.py
src/unitorch/utils/torch_utils.py
src/unitorch/utils/video_utils.py
wiki/configuration.md
wiki/icon.png
wiki/index.md
wiki/installation.md
wiki/overview.md
wiki/overview.png
wiki/benchmarks/index.md
wiki/cli/ast.md
wiki/cli/csv.md
wiki/cli/datatypes.md
wiki/cli/deepspeed.md
wiki/cli/jsonl.md
wiki/cli/parquet.md
wiki/cli/postprocess.md
wiki/cli/preprocess.md
wiki/cli/supervised.md
wiki/cli/models/bart.md
wiki/cli/models/beit.md
wiki/cli/models/bert.md
wiki/cli/models/blip.md
wiki/cli/models/bloom.md
wiki/cli/models/clip.md
wiki/cli/models/deberta.md
wiki/cli/models/diffusers.md
wiki/cli/models/index.md
wiki/cli/models/llama.md
wiki/cli/models/mbart.md
wiki/cli/models/minigpt4.md
wiki/cli/models/mt5.md
wiki/cli/models/peft.md
wiki/cli/models/pegasus.md
wiki/cli/models/roberta.md
wiki/cli/models/swin.md
wiki/cli/models/t5.md
wiki/cli/models/visualbert.md
wiki/cli/models/vit.md
wiki/cli/models/xlm_roberta.md
wiki/cli/models/xpegasus.md
wiki/examples/caption/blip.md
wiki/examples/classification/clip.md
wiki/examples/classification/roberta.md
wiki/examples/classification/swin.md
wiki/examples/diffusion/controlnet.md
wiki/examples/generation/bart.md
wiki/examples/service/zip_image.md
wiki/labs/index.md
wiki/models/bart.md
wiki/models/beit.md
wiki/models/bert.md
wiki/models/blip.md
wiki/models/bloom.md
wiki/models/clip.md
wiki/models/deberta.md
wiki/models/diffusers.md
wiki/models/index.md
wiki/models/llama.md
wiki/models/mbart.md
wiki/models/minigpt4.md
wiki/models/mt5.md
wiki/models/peft.md
wiki/models/pegasus.md
wiki/models/roberta.md
wiki/models/swin.md
wiki/models/t5.md
wiki/models/visualbert.md
wiki/models/vit.md
wiki/models/xlm_roberta.md
wiki/models/xpegasus.md