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/AutoAWQ/gemm_hpu.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/accelerate/utils/transformer_engine.py
optimum/habana/diffusers/__init__.py
optimum/habana/diffusers/models/__init__.py
optimum/habana/diffusers/models/attention_processor.py
optimum/habana/diffusers/models/cogvideox_transformer_3d.py
optimum/habana/diffusers/models/controlnet_sdv.py
optimum/habana/diffusers/models/embeddings.py
optimum/habana/diffusers/models/unet_2d.py
optimum/habana/diffusers/models/unet_2d_condition.py
optimum/habana/diffusers/models/unet_spatio_temporal_condition_controlnet.py
optimum/habana/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py
optimum/habana/diffusers/pipelines/auto_pipeline.py
optimum/habana/diffusers/pipelines/pipeline_utils.py
optimum/habana/diffusers/pipelines/cogvideox/pipeline_cogvideox.py
optimum/habana/diffusers/pipelines/controlnet/pipeline_controlnet.py
optimum/habana/diffusers/pipelines/controlnet/pipeline_stable_video_diffusion_controlnet.py
optimum/habana/diffusers/pipelines/ddpm/pipeline_ddpm.py
optimum/habana/diffusers/pipelines/flux/pipeline_flux.py
optimum/habana/diffusers/pipelines/flux/pipeline_flux_img2img.py
optimum/habana/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py
optimum/habana/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
optimum/habana/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py
optimum/habana/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py
optimum/habana/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py
optimum/habana/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py
optimum/habana/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.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_3/pipeline_stable_diffusion_3.py
optimum/habana/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py
optimum/habana/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py
optimum/habana/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py
optimum/habana/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_mlperf.py
optimum/habana/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py
optimum/habana/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.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/diffusers/schedulers/scheduling_flow_mactch_euler_discrete.py
optimum/habana/diffusers/utils/torch_utils.py
optimum/habana/distributed/__init__.py
optimum/habana/distributed/contextparallel.py
optimum/habana/distributed/distributed_runner.py
optimum/habana/distributed/fast_ddp.py
optimum/habana/distributed/parallel_state.py
optimum/habana/distributed/serialization.py
optimum/habana/distributed/strategy.py
optimum/habana/distributed/tensorparallel.py
optimum/habana/distributed/tp.py
optimum/habana/distributed/tp_wrapping.py
optimum/habana/peft/__init__.py
optimum/habana/peft/layer.py
optimum/habana/peft/peft_model.py
optimum/habana/quantizers/bitsandbytes.py
optimum/habana/sentence_transformers/__init__.py
optimum/habana/sentence_transformers/modeling_utils.py
optimum/habana/sentence_transformers/st_gaudi_data_collator.py
optimum/habana/sentence_transformers/st_gaudi_encoder.py
optimum/habana/sentence_transformers/st_gaudi_trainer.py
optimum/habana/sentence_transformers/st_gaudi_training_args.py
optimum/habana/sentence_transformers/st_gaudi_transformer.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_rope_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/candidate_generator.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/awq.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/baichuan/__init__.py
optimum/habana/transformers/models/baichuan/configuration_baichuan.py
optimum/habana/transformers/models/baichuan/generation_utils.py
optimum/habana/transformers/models/baichuan/modeling_baichuan.py
optimum/habana/transformers/models/baichuan/tokenization_baichuan.py
optimum/habana/transformers/models/bart/__init__.py
optimum/habana/transformers/models/bart/modeling_bart.py
optimum/habana/transformers/models/bert/__init__.py
optimum/habana/transformers/models/bert/modeling_bert.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/chatglm/__init__.py
optimum/habana/transformers/models/chatglm/configuration_chatglm.py
optimum/habana/transformers/models/chatglm/modeling_chatglm.py
optimum/habana/transformers/models/chatglm/tokenization_chatglm.py
optimum/habana/transformers/models/clip/__init__.py
optimum/habana/transformers/models/clip/modeling_clip.py
optimum/habana/transformers/models/codegen/__init__.py
optimum/habana/transformers/models/codegen/modeling_codegen.py
optimum/habana/transformers/models/cohere/__init__.py
optimum/habana/transformers/models/cohere/modeling_cohere.py
optimum/habana/transformers/models/decilm/__init__.py
optimum/habana/transformers/models/decilm/configuration_decilm.py
optimum/habana/transformers/models/decilm/modeling_decilm.py
optimum/habana/transformers/models/deepseek_v2/__init__.py
optimum/habana/transformers/models/deepseek_v2/configuration_deepseek_v2.py
optimum/habana/transformers/models/deepseek_v2/modeling_deepseek_v2.py
optimum/habana/transformers/models/deepseek_v2/tokenization_deepseek_v2.py
optimum/habana/transformers/models/deepseek_v3/__init__.py
optimum/habana/transformers/models/deepseek_v3/configuration_deepseek_v3.py
optimum/habana/transformers/models/deepseek_v3/modeling_deepseek_v3.py
optimum/habana/transformers/models/detr/__init__.py
optimum/habana/transformers/models/detr/modeling_detr.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/falcon_mamba/__init__.py
optimum/habana/transformers/models/falcon_mamba/modeling_falcon_mamba.py
optimum/habana/transformers/models/gemma/__init__.py
optimum/habana/transformers/models/gemma/modeling_gemma.py
optimum/habana/transformers/models/gemma2/__init__.py
optimum/habana/transformers/models/gemma2/modeling_gemma2.py
optimum/habana/transformers/models/glm4v/__init__.py
optimum/habana/transformers/models/glm4v/configuration_chatglm.py
optimum/habana/transformers/models/glm4v/modeling_chatglm.py
optimum/habana/transformers/models/glm4v/tokenization_chatglm.py
optimum/habana/transformers/models/glm4v/visual.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_neo/__init__.py
optimum/habana/transformers/models/gpt_neo/modeling_gpt_neo.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/idefics2/__init__.py
optimum/habana/transformers/models/idefics2/image_processing_idefics2.py
optimum/habana/transformers/models/idefics2/modeling_idefics2.py
optimum/habana/transformers/models/llama/__init__.py
optimum/habana/transformers/models/llama/configuration_llama.py
optimum/habana/transformers/models/llama/modeling_llama.py
optimum/habana/transformers/models/llava/__init__.py
optimum/habana/transformers/models/llava/modeling_llava.py
optimum/habana/transformers/models/llava_next/__init__.py
optimum/habana/transformers/models/llava_next/modeling_llava_next.py
optimum/habana/transformers/models/llava_onevision/__init__.py
optimum/habana/transformers/models/llava_onevision/modeling_llava_onevision.py
optimum/habana/transformers/models/mamba/__init__.py
optimum/habana/transformers/models/mamba/modeling_mamba.py
optimum/habana/transformers/models/minicpm/__init__.py
optimum/habana/transformers/models/minicpm/configuration_minicpm.py
optimum/habana/transformers/models/minicpm/modeling_minicpm.py
optimum/habana/transformers/models/mistral/__init__.py
optimum/habana/transformers/models/mistral/configuration_mistral.py
optimum/habana/transformers/models/mistral/modeling_mistral.py
optimum/habana/transformers/models/mixtral/__init__.py
optimum/habana/transformers/models/mixtral/configuration_mixtral.py
optimum/habana/transformers/models/mixtral/modeling_mixtral.py
optimum/habana/transformers/models/mllama/__init__.py
optimum/habana/transformers/models/mllama/modeling_mllama.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/owlvit/__init__.py
optimum/habana/transformers/models/owlvit/modeling_owlvit.py
optimum/habana/transformers/models/paligemma/__init__.py
optimum/habana/transformers/models/paligemma/modeling_paligemma.py
optimum/habana/transformers/models/persimmon/__init__.py
optimum/habana/transformers/models/persimmon/modeling_persimmon.py
optimum/habana/transformers/models/phi/__init__.py
optimum/habana/transformers/models/phi/modeling_phi.py
optimum/habana/transformers/models/qwen2/__init__.py
optimum/habana/transformers/models/qwen2/modeling_qwen2.py
optimum/habana/transformers/models/qwen2_moe/__init__.py
optimum/habana/transformers/models/qwen2_moe/modeling_qwen2_moe.py
optimum/habana/transformers/models/qwen2_vl/__init__.py
optimum/habana/transformers/models/qwen2_vl/modeling_qwen2_vl.py
optimum/habana/transformers/models/qwen3/__init__.py
optimum/habana/transformers/models/qwen3/modeling_qwen3.py
optimum/habana/transformers/models/qwen3_moe/__init__.py
optimum/habana/transformers/models/qwen3_moe/modeling_qwen3_moe.py
optimum/habana/transformers/models/seamless_m4t/__init__.py
optimum/habana/transformers/models/seamless_m4t/modeling_seamless_m4t.py
optimum/habana/transformers/models/siglip/__init__.py
optimum/habana/transformers/models/siglip/modeling_siglip.py
optimum/habana/transformers/models/speecht5/__init__.py
optimum/habana/transformers/models/speecht5/modeling_speecht5.py
optimum/habana/transformers/models/stablelm/__init__.py
optimum/habana/transformers/models/stablelm/modeling_stablelm.py
optimum/habana/transformers/models/starcoder2/__init__.py
optimum/habana/transformers/models/starcoder2/modeling_starcoder2.py
optimum/habana/transformers/models/t5/__init__.py
optimum/habana/transformers/models/t5/modeling_t5.py
optimum/habana/transformers/models/table_transformer/__init__.py
optimum/habana/transformers/models/table_transformer/modeling_table_transformer.py
optimum/habana/transformers/models/video_llava/__init__.py
optimum/habana/transformers/models/video_llava/modeling_video_llava.py
optimum/habana/transformers/models/video_llava/processing_video_llava.py
optimum/habana/transformers/models/vision_encoder_decoder/__init__.py
optimum/habana/transformers/models/vision_encoder_decoder/modeling_vision_encoder_decoder.py
optimum/habana/transformers/models/vit/__init__.py
optimum/habana/transformers/models/vit/modeling_vit.py
optimum/habana/transformers/models/vits/__init__.py
optimum/habana/transformers/models/vits/modeling_vits.py
optimum/habana/transformers/models/wav2vec2/__init__.py
optimum/habana/transformers/models/wav2vec2/modeling_wav2vec2.py
optimum/habana/transformers/models/whisper/__init__.py
optimum/habana/transformers/models/whisper/modeling_whisper.py
optimum/habana/transformers/models/xglm/__init__.py
optimum/habana/transformers/models/xglm/modeling_xglm.py
optimum/habana/transformers/models/xlm_roberta/__init__.py
optimum/habana/transformers/models/xlm_roberta/modeling_xlm_roberta.py
optimum/habana/trl/__init__.py
optimum/habana/trl/models/__init__.py
optimum/habana/trl/models/modeling_base.py
optimum/habana/trl/models/modeling_sd_base.py
optimum/habana/trl/trainer/__init__.py
optimum/habana/trl/trainer/ddpo_trainer.py
optimum/habana/trl/trainer/dpo_config.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_config.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_bnb_inference.py
tests/test_bnb_qlora.py
tests/test_custom_file_input.py
tests/test_diffusers.py
tests/test_encoder_decoder.py
tests/test_examples.py
tests/test_examples_match_transformers.py
tests/test_feature_extraction.py
tests/test_fp8_examples.py
tests/test_fsdp_examples.py
tests/test_functional_text_generation_example.py
tests/test_gaudi_configuration.py
tests/test_image_classification.py
tests/test_image_segmentation.py
tests/test_image_to_text_example.py
tests/test_object_detection.py
tests/test_object_segmentation.py
tests/test_openclip_vqa.py
tests/test_peft_inference.py
tests/test_pipeline.py
tests/test_sentence_transformers.py
tests/test_table_transformer.py
tests/test_text_generation_example.py
tests/test_timm.py
tests/test_trainer.py
tests/test_trainer_distributed.py
tests/test_trainer_seq2seq.py
tests/test_trl.py
tests/test_video_llava.py
tests/test_video_mae.py
tests/test_zero_shot_object_detection.py