LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.cfg
setup.py
optimum/habana/__init__.py
optimum/habana/checkpoint_utils.py
optimum/habana/utils.py
optimum/habana/version.py
optimum/habana/accelerate/__init__.py
optimum/habana/accelerate/accelerator.py
optimum/habana/accelerate/data_loader.py
optimum/habana/accelerate/state.py
optimum/habana/accelerate/utils/__init__.py
optimum/habana/accelerate/utils/dataclasses.py
optimum/habana/accelerate/utils/operations.py
optimum/habana/diffusers/__init__.py
optimum/habana/diffusers/models/__init__.py
optimum/habana/diffusers/models/unet_2d_condition.py
optimum/habana/diffusers/pipelines/pipeline_utils.py
optimum/habana/diffusers/pipelines/controlnet/pipeline_controlnet.py
optimum/habana/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
optimum/habana/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_ldm3d.py
optimum/habana/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py
optimum/habana/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py
optimum/habana/diffusers/schedulers/__init__.py
optimum/habana/diffusers/schedulers/scheduling_ddim.py
optimum/habana/diffusers/schedulers/scheduling_euler_ancestral_discrete.py
optimum/habana/diffusers/schedulers/scheduling_euler_discrete.py
optimum/habana/distributed/__init__.py
optimum/habana/distributed/distributed_runner.py
optimum/habana/distributed/fast_ddp.py
optimum/habana/peft/__init__.py
optimum/habana/peft/layer.py
optimum/habana/transformers/__init__.py
optimum/habana/transformers/gaudi_configuration.py
optimum/habana/transformers/gradient_checkpointing.py
optimum/habana/transformers/modeling_attn_mask_utils.py
optimum/habana/transformers/modeling_utils.py
optimum/habana/transformers/trainer.py
optimum/habana/transformers/trainer_seq2seq.py
optimum/habana/transformers/trainer_utils.py
optimum/habana/transformers/training_args.py
optimum/habana/transformers/training_args_seq2seq.py
optimum/habana/transformers/generation/__init__.py
optimum/habana/transformers/generation/configuration_utils.py
optimum/habana/transformers/generation/stopping_criteria.py
optimum/habana/transformers/generation/utils.py
optimum/habana/transformers/integrations/deepspeed.py
optimum/habana/transformers/models/__init__.py
optimum/habana/transformers/models/modeling_all_models.py
optimum/habana/transformers/models/albert/__init__.py
optimum/habana/transformers/models/albert/modeling_albert.py
optimum/habana/transformers/models/bart/__init__.py
optimum/habana/transformers/models/bart/modeling_bart.py
optimum/habana/transformers/models/blip/__init__.py
optimum/habana/transformers/models/blip/modeling_blip.py
optimum/habana/transformers/models/blip/modeling_blip_text.py
optimum/habana/transformers/models/bloom/__init__.py
optimum/habana/transformers/models/bloom/modeling_bloom.py
optimum/habana/transformers/models/codegen/__init__.py
optimum/habana/transformers/models/codegen/modeling_codegen.py
optimum/habana/transformers/models/esm/__init__.py
optimum/habana/transformers/models/esm/modeling_esmfold.py
optimum/habana/transformers/models/falcon/__init__.py
optimum/habana/transformers/models/falcon/modeling_falcon.py
optimum/habana/transformers/models/gpt2/__init__.py
optimum/habana/transformers/models/gpt2/modeling_gpt2.py
optimum/habana/transformers/models/gpt_bigcode/__init__.py
optimum/habana/transformers/models/gpt_bigcode/modeling_gpt_bigcode.py
optimum/habana/transformers/models/gpt_neox/__init__.py
optimum/habana/transformers/models/gpt_neox/modeling_gpt_neox.py
optimum/habana/transformers/models/gptj/__init__.py
optimum/habana/transformers/models/gptj/modeling_gptj.py
optimum/habana/transformers/models/llama/__init__.py
optimum/habana/transformers/models/llama/modeling_llama.py
optimum/habana/transformers/models/mistral/__init__.py
optimum/habana/transformers/models/mistral/modeling_mistral.py
optimum/habana/transformers/models/mixtral/__init__.py
optimum/habana/transformers/models/mixtral/modeling_mixtral.py
optimum/habana/transformers/models/mpt/__init__.py
optimum/habana/transformers/models/mpt/modeling_mpt.py
optimum/habana/transformers/models/opt/__init__.py
optimum/habana/transformers/models/opt/modeling_opt.py
optimum/habana/transformers/models/phi/__init__.py
optimum/habana/transformers/models/phi/modeling_phi.py
optimum/habana/transformers/models/speecht5/__init__.py
optimum/habana/transformers/models/speecht5/modeling_speecht5.py
optimum/habana/transformers/models/swin/__init__.py
optimum/habana/transformers/models/swin/modeling_swin.py
optimum/habana/transformers/models/t5/__init__.py
optimum/habana/transformers/models/t5/modeling_t5.py
optimum/habana/transformers/models/vit/__init__.py
optimum/habana/transformers/models/vit/modeling_vit.py
optimum/habana/transformers/models/wav2vec2/__init__.py
optimum/habana/transformers/models/wav2vec2/modeling_wav2vec2.py
optimum/habana/trl/__init__.py
optimum/habana/trl/models/__init__.py
optimum/habana/trl/models/modeling_base.py
optimum/habana/trl/trainer/__init__.py
optimum/habana/trl/trainer/dpo_trainer.py
optimum/habana/trl/trainer/ppo_config.py
optimum/habana/trl/trainer/ppo_trainer.py
optimum/habana/trl/trainer/reward_trainer.py
optimum/habana/trl/trainer/sft_trainer.py
optimum_habana.egg-info/PKG-INFO
optimum_habana.egg-info/SOURCES.txt
optimum_habana.egg-info/dependency_links.txt
optimum_habana.egg-info/not-zip-safe
optimum_habana.egg-info/requires.txt
optimum_habana.egg-info/top_level.txt
tests/test_diffusers.py
tests/test_encoder_decoder.py
tests/test_examples.py
tests/test_examples_match_transformers.py
tests/test_fsdp_examples.py
tests/test_gaudi_configuration.py
tests/test_text_generation_example.py
tests/test_trainer.py
tests/test_trainer_distributed.py
tests/test_trainer_seq2seq.py