LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.cfg
setup.py
optimum/habana/__init__.py
optimum/habana/utils.py
optimum/habana/version.py
optimum/habana/accelerate/__init__.py
optimum/habana/accelerate/accelerator.py
optimum/habana/accelerate/state.py
optimum/habana/accelerate/utils/__init__.py
optimum/habana/accelerate/utils/dataclasses.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/stable_diffusion/pipeline_stable_diffusion.py
optimum/habana/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_ldm3d.py
optimum/habana/diffusers/schedulers/__init__.py
optimum/habana/diffusers/schedulers/scheduling_ddim.py
optimum/habana/distributed/__init__.py
optimum/habana/distributed/distributed_runner.py
optimum/habana/distributed/fast_ddp.py
optimum/habana/transformers/__init__.py
optimum/habana/transformers/deepspeed.py
optimum/habana/transformers/gaudi_configuration.py
optimum/habana/transformers/gradient_checkpointing.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/utils.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/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_esm.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/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/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.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_examples.py
tests/test_examples_match_transformers.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