├── .github ├── CODEOWNERS └── PULL_REQUEST_TEMPLATE.md ├── .gitignore ├── .gitmodules ├── CONTRIBUTING.md ├── PyTorch ├── __init__.py ├── computer_vision │ ├── classification │ │ └── torchvision │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── data_loaders.py │ │ │ ├── googlenet_utils.py │ │ │ ├── inference.py │ │ │ ├── main.py │ │ │ ├── media_pipe_settings.py │ │ │ ├── model │ │ │ ├── __init__.py │ │ │ ├── optimizer.py │ │ │ ├── resnet.py │ │ │ └── utils.py │ │ │ ├── ops_bf16_Resnet.txt │ │ │ ├── ops_fp32_Resnet.txt │ │ │ ├── requirements.txt │ │ │ ├── requirements_u24.txt │ │ │ ├── resnet_media_pipe.py │ │ │ ├── train.py │ │ │ └── utils.py │ ├── detection │ │ └── yolox │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── assets │ │ │ ├── demo.png │ │ │ ├── dog.jpg │ │ │ ├── git_fig.png │ │ │ └── logo.png │ │ │ ├── demo │ │ │ ├── MegEngine │ │ │ │ ├── cpp │ │ │ │ │ ├── README.md │ │ │ │ │ ├── build.sh │ │ │ │ │ └── yolox.cpp │ │ │ │ └── python │ │ │ │ │ ├── README.md │ │ │ │ │ ├── build.py │ │ │ │ │ ├── convert_weights.py │ │ │ │ │ ├── demo.py │ │ │ │ │ ├── dump.py │ │ │ │ │ └── models │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── darknet.py │ │ │ │ │ ├── network_blocks.py │ │ │ │ │ ├── yolo_fpn.py │ │ │ │ │ ├── yolo_head.py │ │ │ │ │ ├── yolo_pafpn.py │ │ │ │ │ └── yolox.py │ │ │ ├── ONNXRuntime │ │ │ │ ├── README.md │ │ │ │ └── onnx_inference.py │ │ │ ├── OpenVINO │ │ │ │ ├── README.md │ │ │ │ ├── cpp │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ ├── README.md │ │ │ │ │ └── yolox_openvino.cpp │ │ │ │ └── python │ │ │ │ │ ├── README.md │ │ │ │ │ └── openvino_inference.py │ │ │ ├── TensorRT │ │ │ │ ├── cpp │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ ├── README.md │ │ │ │ │ ├── logging.h │ │ │ │ │ └── yolox.cpp │ │ │ │ └── python │ │ │ │ │ └── README.md │ │ │ └── ncnn │ │ │ │ ├── android │ │ │ │ ├── README.md │ │ │ │ ├── app │ │ │ │ │ ├── build.gradle │ │ │ │ │ └── src │ │ │ │ │ │ └── main │ │ │ │ │ │ ├── AndroidManifest.xml │ │ │ │ │ │ ├── assets │ │ │ │ │ │ └── yolox.param │ │ │ │ │ │ ├── java │ │ │ │ │ │ └── com │ │ │ │ │ │ │ └── megvii │ │ │ │ │ │ │ └── yoloXncnn │ │ │ │ │ │ │ ├── MainActivity.java │ │ │ │ │ │ │ ├── YOLOXncnn.java │ │ │ │ │ │ │ └── yoloXncnn.java │ │ │ │ │ │ ├── jni │ │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ │ └── yoloXncnn_jni.cpp │ │ │ │ │ │ └── res │ │ │ │ │ │ ├── layout │ │ │ │ │ │ └── main.xml │ │ │ │ │ │ └── values │ │ │ │ │ │ └── strings.xml │ │ │ │ ├── build.gradle │ │ │ │ ├── gradle │ │ │ │ │ └── wrapper │ │ │ │ │ │ ├── gradle-wrapper.jar │ │ │ │ │ │ └── gradle-wrapper.properties │ │ │ │ ├── gradlew │ │ │ │ ├── gradlew.bat │ │ │ │ └── settings.gradle │ │ │ │ └── cpp │ │ │ │ ├── README.md │ │ │ │ └── yolox.cpp │ │ │ ├── docs │ │ │ ├── .gitignore │ │ │ ├── Makefile │ │ │ ├── _static │ │ │ │ └── css │ │ │ │ │ └── custom.css │ │ │ ├── conf.py │ │ │ ├── demo │ │ │ │ ├── megengine_cpp_readme.md │ │ │ │ ├── megengine_py_readme.md │ │ │ │ ├── ncnn_android_readme.md │ │ │ │ ├── ncnn_cpp_readme.md │ │ │ │ ├── onnx_readme.md │ │ │ │ ├── openvino_cpp_readme.md │ │ │ │ ├── openvino_py_readme.md │ │ │ │ ├── trt_cpp_readme.md │ │ │ │ └── trt_py_readme.md │ │ │ ├── index.rst │ │ │ ├── manipulate_training_image_size.md │ │ │ ├── model_zoo.md │ │ │ ├── quick_run.md │ │ │ ├── requirements-doc.txt │ │ │ ├── train_custom_data.md │ │ │ └── updates_note.md │ │ │ ├── download_dataset.sh │ │ │ ├── exps │ │ │ ├── default │ │ │ │ ├── __init__.py │ │ │ │ ├── yolov3.py │ │ │ │ ├── yolox_l.py │ │ │ │ ├── yolox_m.py │ │ │ │ ├── yolox_nano.py │ │ │ │ ├── yolox_s.py │ │ │ │ ├── yolox_tiny.py │ │ │ │ └── yolox_x.py │ │ │ └── example │ │ │ │ ├── custom │ │ │ │ ├── nano.py │ │ │ │ └── yolox_s.py │ │ │ │ └── yolox_voc │ │ │ │ └── yolox_voc_s.py │ │ │ ├── hubconf.py │ │ │ ├── ops_bf16_yolox.txt │ │ │ ├── ops_fp32_yolox.txt │ │ │ ├── requirements.txt │ │ │ ├── setup.cfg │ │ │ ├── setup.py │ │ │ ├── tests │ │ │ ├── __init__.py │ │ │ └── utils │ │ │ │ └── test_model_utils.py │ │ │ ├── tools │ │ │ ├── __init__.py │ │ │ ├── demo.py │ │ │ ├── eval.py │ │ │ ├── export_onnx.py │ │ │ ├── export_torchscript.py │ │ │ ├── train.py │ │ │ └── trt.py │ │ │ └── yolox │ │ │ ├── __init__.py │ │ │ ├── core │ │ │ ├── __init__.py │ │ │ ├── launch.py │ │ │ └── trainer.py │ │ │ ├── data │ │ │ ├── __init__.py │ │ │ ├── data_augment.py │ │ │ ├── data_prefetcher.py │ │ │ ├── dataloading.py │ │ │ ├── dataloading_hpu_loader.py │ │ │ ├── dataloading_hpu_mediapipe.py │ │ │ ├── datasets │ │ │ │ ├── __init__.py │ │ │ │ ├── coco.py │ │ │ │ ├── coco_classes.py │ │ │ │ ├── datasets_wrapper.py │ │ │ │ ├── mosaicdetection.py │ │ │ │ ├── voc.py │ │ │ │ └── voc_classes.py │ │ │ └── samplers.py │ │ │ ├── evaluators │ │ │ ├── __init__.py │ │ │ ├── coco_evaluator.py │ │ │ ├── voc_eval.py │ │ │ └── voc_evaluator.py │ │ │ ├── exp │ │ │ ├── __init__.py │ │ │ ├── base_exp.py │ │ │ ├── build.py │ │ │ ├── default │ │ │ │ └── __init__.py │ │ │ └── yolox_base.py │ │ │ ├── layers │ │ │ ├── __init__.py │ │ │ ├── csrc │ │ │ │ ├── cocoeval │ │ │ │ │ ├── cocoeval.cpp │ │ │ │ │ └── cocoeval.h │ │ │ │ └── vision.cpp │ │ │ └── fast_coco_eval_api.py │ │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── build.py │ │ │ ├── darknet.py │ │ │ ├── losses.py │ │ │ ├── network_blocks.py │ │ │ ├── yolo_fpn.py │ │ │ ├── yolo_head.py │ │ │ ├── yolo_head_script.py │ │ │ ├── yolo_pafpn.py │ │ │ └── yolox.py │ │ │ ├── tools │ │ │ └── __init__.py │ │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── allreduce_norm.py │ │ │ ├── boxes.py │ │ │ ├── checkpoint.py │ │ │ ├── compat.py │ │ │ ├── demo_utils.py │ │ │ ├── dist.py │ │ │ ├── ema.py │ │ │ ├── logger.py │ │ │ ├── lr_scheduler.py │ │ │ ├── metric.py │ │ │ ├── model_utils.py │ │ │ ├── setup_env.py │ │ │ └── visualize.py │ └── segmentation │ │ └── Unet │ │ ├── Dockerfile │ │ ├── LICENSE │ │ ├── README.md │ │ ├── config │ │ ├── ops_bf16_unet.txt │ │ └── ops_fp32_unet.txt │ │ ├── data_loading │ │ ├── dali_loader.py │ │ └── data_module.py │ │ ├── data_preprocessing │ │ ├── configs.py │ │ └── preprocessor.py │ │ ├── download.py │ │ ├── evaluate.py │ │ ├── images │ │ └── unet3d.png │ │ ├── lightning_trainer │ │ ├── __init__.py │ │ └── ptl.py │ │ ├── main.py │ │ ├── models │ │ ├── dice.py │ │ ├── layers.py │ │ ├── loss.py │ │ ├── metrics.py │ │ ├── monai_sliding_window_inference.py │ │ ├── nn_unet.py │ │ ├── pl_metric.py │ │ └── unet.py │ │ ├── preprocess.py │ │ ├── pytorch │ │ ├── early_stopping_unet.py │ │ ├── misc.py │ │ ├── npt.py │ │ └── trainer.py │ │ ├── requirements.txt │ │ ├── requirements_u22.txt │ │ ├── requirements_u24.txt │ │ ├── scripts │ │ ├── benchmark.py │ │ ├── inference.py │ │ └── train.py │ │ └── utils │ │ ├── __init__.py │ │ ├── early_stopping_unet.py │ │ ├── gpu_affinity.py │ │ ├── logger.py │ │ └── utils.py ├── examples │ ├── DeepSpeed │ │ └── cifar_example │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── cifar10_deepspeed.py │ │ │ ├── ds_config.json │ │ │ ├── requirements.txt │ │ │ └── run_ds_habanax8.sh │ ├── bucketing │ │ ├── README.md │ │ ├── brute_force_min_pad_waste.py │ │ ├── bucket.py │ │ ├── bucket_analysis.svg │ │ ├── bucket_analysis_bar_gaussian.svg │ │ ├── bucket_analysis_bar_squad.svg │ │ ├── bucket_analysis_num_steps_gaussian.svg │ │ ├── datasets_library.py │ │ ├── gaussian.svg │ │ ├── lloyd_max_bucket.py │ │ ├── plotting.py │ │ ├── requirements.txt │ │ ├── run_demo_bucketing_gaussian.py │ │ ├── run_demo_bucketing_squad.py │ │ ├── run_demo_controlling_num_steps.py │ │ ├── run_demo_gaussian.py │ │ ├── run_demo_squad.py │ │ ├── squad.svg │ │ └── test.py │ ├── computer_vision │ │ └── hello_world │ │ │ ├── README.md │ │ │ ├── example.py │ │ │ ├── mnist.py │ │ │ └── utils.py │ ├── custom_op │ │ ├── custom_fusedsdpa │ │ │ ├── README.md │ │ │ └── custom_fusedsdpa_op.patch │ │ ├── legacy_custom_op_API │ │ │ ├── custom_relu │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── custom_relu.py │ │ │ │ ├── custom_relu_op.patch │ │ │ │ ├── hpu_custom_op_relu_test.py │ │ │ │ ├── hpu_custom_relu.cpp │ │ │ │ └── setup.py │ │ │ └── custom_topk │ │ │ │ ├── README.md │ │ │ │ ├── hpu_custom_op_topk_test.py │ │ │ │ ├── hpu_custom_topk.cpp │ │ │ │ └── setup.py │ │ └── pt2_custom_op_API │ │ │ ├── custom_relu │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── custom_relu.py │ │ │ ├── custom_relu_op.patch │ │ │ ├── hpu_custom_op_relu_test.py │ │ │ ├── hpu_custom_relu.cpp │ │ │ └── setup.py │ │ │ └── custom_topk │ │ │ ├── README.md │ │ │ ├── hpu_custom_topk.cpp │ │ │ ├── setup.py │ │ │ └── test_hpu_custom_op_topk.py │ ├── gpu_migration │ │ ├── README.md │ │ ├── computer_vision │ │ │ └── classification │ │ │ │ └── torchvision │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── gpu_migration_logs │ │ │ │ └── gpu_migration_958.log │ │ │ │ ├── patches │ │ │ │ ├── lr_scheduler.diff │ │ │ │ ├── minimal_changes.diff │ │ │ │ └── performance_improvements.diff │ │ │ │ ├── presets.py │ │ │ │ ├── sampler.py │ │ │ │ ├── train.py │ │ │ │ ├── train_quantization.py │ │ │ │ ├── transforms.py │ │ │ │ └── utils.py │ │ ├── generative_models │ │ │ └── stable-diffusion │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ ├── autoencoder │ │ │ │ │ ├── autoencoder_kl_16x16x16.yaml │ │ │ │ │ ├── autoencoder_kl_32x32x4.yaml │ │ │ │ │ ├── autoencoder_kl_64x64x3.yaml │ │ │ │ │ └── autoencoder_kl_8x8x64.yaml │ │ │ │ ├── latent-diffusion │ │ │ │ │ ├── celebahq-ldm-vq-4.yaml │ │ │ │ │ ├── cin-ldm-vq-f8.yaml │ │ │ │ │ ├── cin256-v2.yaml │ │ │ │ │ ├── ffhq-ldm-vq-4.yaml │ │ │ │ │ ├── lsun_bedrooms-ldm-vq-4.yaml │ │ │ │ │ ├── lsun_churches-ldm-kl-8.yaml │ │ │ │ │ └── txt2img-1p4B-eval.yaml │ │ │ │ └── stable-diffusion │ │ │ │ │ ├── dev.yaml │ │ │ │ │ ├── dev_mn.yaml │ │ │ │ │ ├── dev_mn_dummy.yaml │ │ │ │ │ ├── inpainting │ │ │ │ │ ├── v1-edgeinpainting.yaml │ │ │ │ │ ├── v1-finetune-for-inpainting-laion-aesthetic-larger-masks-and-ucfg.yaml │ │ │ │ │ ├── v1-finetune-for-inpainting-laion-aesthetic-larger-masks.yaml │ │ │ │ │ └── v1-finetune-for-inpainting-laion-iaesthe.yaml │ │ │ │ │ ├── txt2img-1p4B-multinode-clip-encoder-high-res-512.yaml │ │ │ │ │ ├── txt2img-1p4B-multinode-clip-encoder.yaml │ │ │ │ │ ├── txt2img-1p4B-multinode-t5-encoder.yaml │ │ │ │ │ ├── txt2img-1p4B-multinode.yaml │ │ │ │ │ ├── txt2img-clip-encoder-dev.yaml │ │ │ │ │ ├── txt2img-ldm-frozen-dev.yaml │ │ │ │ │ ├── txt2img-ldm-unfrozen-dev.yaml │ │ │ │ │ ├── txt2img-ldm-vae-f8.yaml │ │ │ │ │ ├── txt2img-multinode-clip-encoder-f16-1024-laion-hr.yaml │ │ │ │ │ ├── txt2img-multinode-clip-encoder-f16-256-pretraining.yaml │ │ │ │ │ ├── txt2img-multinode-clip-encoder-f16-768-laion-hr-inference.yaml │ │ │ │ │ ├── txt2img-multinode-clip-encoder-f16-768-laion-hr.yaml │ │ │ │ │ ├── txt2img-multinode-clip-encoder-f16-768.yaml │ │ │ │ │ ├── txt2img-t5-encoder-dev.yaml │ │ │ │ │ ├── txt2img-upscale-clip-encoder-f16-1024-dev.yaml │ │ │ │ │ ├── txt2img-upscale-clip-encoder-f16-1024.yaml │ │ │ │ │ ├── txt2img-v2-clip-encoder-improved_aesthetics-256-dev.yaml │ │ │ │ │ ├── txt2img-v2-clip-encoder-improved_aesthetics-256.yaml │ │ │ │ │ ├── txt2img-v2-clip-encoder-improved_aesthetics-512-dev.yaml │ │ │ │ │ ├── txt2img-v2-clip-encoder-improved_aesthetics-512.yaml │ │ │ │ │ ├── upscaling │ │ │ │ │ └── upscale-v1-with-f16.yaml │ │ │ │ │ ├── v1-inference.yaml │ │ │ │ │ ├── v1_improvedaesthetics.yaml │ │ │ │ │ ├── v1_laionhr.yaml │ │ │ │ │ ├── v2_laionhr1024.yaml │ │ │ │ │ ├── v2_laionhr1024_2.yaml │ │ │ │ │ ├── v2_pretraining.yaml │ │ │ │ │ └── v3_pretraining.yaml │ │ │ │ ├── environment.yaml │ │ │ │ ├── gpu_migration_logs │ │ │ │ └── gpu_migration_1762.log │ │ │ │ ├── hpu_graph_utils.py │ │ │ │ ├── ldm │ │ │ │ ├── data │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base.py │ │ │ │ │ ├── coco.py │ │ │ │ │ ├── dummy.py │ │ │ │ │ ├── imagenet.py │ │ │ │ │ ├── inpainting │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── synthetic_mask.py │ │ │ │ │ ├── laion.py │ │ │ │ │ └── lsun.py │ │ │ │ ├── lr_scheduler.py │ │ │ │ ├── models │ │ │ │ │ ├── autoencoder.py │ │ │ │ │ └── diffusion │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── classifier.py │ │ │ │ │ │ ├── ddim.py │ │ │ │ │ │ ├── ddpm.py │ │ │ │ │ │ ├── plms.py │ │ │ │ │ │ └── sampling_util.py │ │ │ │ ├── modules │ │ │ │ │ ├── attention.py │ │ │ │ │ ├── diffusionmodules │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── model.py │ │ │ │ │ │ ├── openaimodel.py │ │ │ │ │ │ └── util.py │ │ │ │ │ ├── distributions │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── distributions.py │ │ │ │ │ ├── ema.py │ │ │ │ │ ├── encoders │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── modules.py │ │ │ │ │ ├── evaluate │ │ │ │ │ │ ├── adm_evaluator.py │ │ │ │ │ │ ├── evaluate_perceptualsim.py │ │ │ │ │ │ ├── frechet_video_distance.py │ │ │ │ │ │ ├── ssim.py │ │ │ │ │ │ └── torch_frechet_video_distance.py │ │ │ │ │ ├── image_degradation │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── bsrgan.py │ │ │ │ │ │ ├── bsrgan_light.py │ │ │ │ │ │ ├── utils │ │ │ │ │ │ │ └── test.png │ │ │ │ │ │ └── utils_image.py │ │ │ │ │ ├── losses │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── contperceptual.py │ │ │ │ │ │ └── vqperceptual.py │ │ │ │ │ └── x_transformer.py │ │ │ │ └── util.py │ │ │ │ ├── main.py │ │ │ │ ├── models │ │ │ │ ├── first_stage_models │ │ │ │ │ ├── kl-f16 │ │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── kl-f32 │ │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── kl-f4 │ │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── kl-f8 │ │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── vq-f16 │ │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── vq-f4-noattn │ │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── vq-f4 │ │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── vq-f8-n256 │ │ │ │ │ │ └── config.yaml │ │ │ │ │ └── vq-f8 │ │ │ │ │ │ └── config.yaml │ │ │ │ └── ldm │ │ │ │ │ ├── bsr_sr │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── celeba256 │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── cin256 │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── ffhq256 │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── inpainting_big │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── layout2img-openimages256 │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── lsun_beds256 │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── lsun_churches256 │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── semantic_synthesis256 │ │ │ │ │ └── config.yaml │ │ │ │ │ ├── semantic_synthesis512 │ │ │ │ │ └── config.yaml │ │ │ │ │ └── text2img256 │ │ │ │ │ └── config.yaml │ │ │ │ ├── notebook_helpers.py │ │ │ │ ├── patches │ │ │ │ ├── hpu_graph.diff │ │ │ │ ├── minimal_changes.diff │ │ │ │ └── randn_to_cpu.diff │ │ │ │ ├── requirements.txt │ │ │ │ ├── scripts │ │ │ │ ├── autoencoder-eval.py │ │ │ │ ├── checker.py │ │ │ │ ├── cmd_on_new_ckpt.py │ │ │ │ ├── demo │ │ │ │ │ └── inpainting.py │ │ │ │ ├── download_first_stages.sh │ │ │ │ ├── download_models.sh │ │ │ │ ├── img2img.py │ │ │ │ ├── inpaint.py │ │ │ │ ├── logging_template.py │ │ │ │ ├── mnist-distributed.py │ │ │ │ ├── printckpt.py │ │ │ │ ├── prompts │ │ │ │ │ ├── aesthetic-prompts-plain.txt │ │ │ │ │ ├── aesthetic-prompts-surrealism.txt │ │ │ │ │ ├── prompts-with-wings.txt │ │ │ │ │ ├── six-prompts │ │ │ │ │ ├── weird-dalle-prompts.txt │ │ │ │ │ ├── wings1.txt │ │ │ │ │ ├── wings2.txt │ │ │ │ │ ├── wings3.txt │ │ │ │ │ └── wings4.txt │ │ │ │ ├── prune-ckpt.py │ │ │ │ ├── sample_diffusion.py │ │ │ │ ├── slurm │ │ │ │ │ ├── README.md │ │ │ │ │ ├── resume_512 │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── resume_512_improvedaesthetic │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── resume_768_hr │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v1-upscaling-f16-pretraining-512-aesthetics │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v1_edgeinpainting │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v1_iahr_torch111 │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v1_iahr_torch111_ucg │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v1_improvedaesthetics │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v1_improvedaesthetics_torch111 │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v1_inpainting_aesthetics-larger-masks-ucg │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v1_inpainting_aesthetics-larger-masks │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v1_inpainting_improvedaesthetics_torch111 │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v1_laionhr_torch111 │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v2_laionhr1024 │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v2_laionhr1024_2 │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ ├── v2_pretraining │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ │ └── v3_pretraining │ │ │ │ │ │ ├── launcher.sh │ │ │ │ │ │ └── sbatch.sh │ │ │ │ ├── test_gpu.py │ │ │ │ ├── test_gpu.sh │ │ │ │ ├── txt2img.py │ │ │ │ └── vqgan_codebook_visualizer.py │ │ │ │ └── setup.py │ │ ├── nlp │ │ │ ├── DeepSpeedExamples │ │ │ │ └── Megatron-DeepSpeed │ │ │ │ │ ├── CODEOWNERS │ │ │ │ │ ├── LICENSE │ │ │ │ │ ├── MANIFEST.in │ │ │ │ │ ├── README.md │ │ │ │ │ ├── SECURITY.md │ │ │ │ │ ├── dataset │ │ │ │ │ ├── README.md │ │ │ │ │ ├── download_books.sh │ │ │ │ │ ├── download_ckpt.sh │ │ │ │ │ └── download_vocab.sh │ │ │ │ │ ├── examples │ │ │ │ │ ├── MoE │ │ │ │ │ │ ├── ds_config_gpt_TEMPLATE.json │ │ │ │ │ │ ├── ds_config_gpt_Zero2_TEMPLATE.json │ │ │ │ │ │ ├── ds_evalharness.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_1.3B_MoE128.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_1.3B_PR-MoE64or128.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_1.3B_PR-MoE64or128_MoS.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_1.3B_dense.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_1.3B_dense_cl.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_125M_MoE64.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_125M_dense_cl.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_350M_MoE128.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_350M_PR-MoE32or64.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_350M_PR-MoE32or64_MoS.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_350M_dense.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_6.7B_dense.sh │ │ │ │ │ │ └── readme_evalharness.md │ │ │ │ │ ├── README.md │ │ │ │ │ ├── azure │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── run-175b.sh │ │ │ │ │ │ ├── run-1t.sh │ │ │ │ │ │ └── run-benchmark-model.sh │ │ │ │ │ ├── azureml │ │ │ │ │ │ ├── Dockerfile.dockerfile │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── aml_submit.py │ │ │ │ │ │ └── prepare_dataset.py │ │ │ │ │ ├── compression │ │ │ │ │ │ ├── 125M-Int8-test-64gpu-distilled-group48.sh │ │ │ │ │ │ ├── 125M-L10-Int8-test-64gpu-distilled-group48.sh │ │ │ │ │ │ ├── 125M-L12-Int8-test-64gpu-distilled-group48.sh │ │ │ │ │ │ ├── ds_config_gpt_TEMPLATE.json │ │ │ │ │ │ ├── ds_config_gpt_TEMPLATE_compression.json │ │ │ │ │ │ ├── ds_evalharness.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_1.3B_dense_cl_kd.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_125M_dense_cl_kd.sh │ │ │ │ │ │ ├── ds_pretrain_gpt_125M_dense_kd.sh │ │ │ │ │ │ └── ds_pretrain_gpt_350M_dense_kd.sh │ │ │ │ │ ├── create_embeddings.sh │ │ │ │ │ ├── curriculum_learning │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── ds_pretrain_gpt2.sh │ │ │ │ │ │ ├── ds_train.sh │ │ │ │ │ │ ├── ds_zero_stage_1_config_baseline.json │ │ │ │ │ │ └── ds_zero_stage_1_config_curriculum_fixed_linear.json │ │ │ │ │ ├── evaluate_ict_zeroshot_nq.sh │ │ │ │ │ ├── evaluate_zeroshot_gpt.sh │ │ │ │ │ ├── finetune_mnli_distributed.sh │ │ │ │ │ ├── finetune_race_distributed.sh │ │ │ │ │ ├── generate_text.sh │ │ │ │ │ ├── merge_mp_bert.sh │ │ │ │ │ ├── pretrain_bert.sh │ │ │ │ │ ├── pretrain_bert_distributed.sh │ │ │ │ │ ├── pretrain_bert_distributed_with_mp.sh │ │ │ │ │ ├── pretrain_gpt.sh │ │ │ │ │ ├── pretrain_gpt3_175B.sh │ │ │ │ │ ├── pretrain_gpt_distributed.sh │ │ │ │ │ ├── pretrain_gpt_distributed_with_mp.sh │ │ │ │ │ ├── pretrain_ict.sh │ │ │ │ │ ├── pretrain_t5.sh │ │ │ │ │ ├── pretrain_t5_distributed.sh │ │ │ │ │ ├── pretrain_t5_distributed_with_mp.sh │ │ │ │ │ └── run_deepspeed_example.sh │ │ │ │ │ ├── gpu_migration_logs │ │ │ │ │ └── gpu_migration_424488.log │ │ │ │ │ ├── images │ │ │ │ │ └── cases_april2021.png │ │ │ │ │ ├── megatron │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── arguments.py │ │ │ │ │ ├── checkpointing.py │ │ │ │ │ ├── data │ │ │ │ │ │ ├── Makefile │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── autoaugment.py │ │ │ │ │ │ ├── bert_dataset.py │ │ │ │ │ │ ├── biencoder_dataset_utils.py │ │ │ │ │ │ ├── blendable_dataset.py │ │ │ │ │ │ ├── data_samplers.py │ │ │ │ │ │ ├── dataset_utils.py │ │ │ │ │ │ ├── gpt_dataset.py │ │ │ │ │ │ ├── helpers.cpp │ │ │ │ │ │ ├── ict_dataset.py │ │ │ │ │ │ ├── indexed_dataset.py │ │ │ │ │ │ ├── orqa_wiki_dataset.py │ │ │ │ │ │ ├── realm_dataset_utils.py │ │ │ │ │ │ ├── realm_index.py │ │ │ │ │ │ ├── t5_dataset.py │ │ │ │ │ │ ├── test │ │ │ │ │ │ │ ├── test_indexed_dataset.py │ │ │ │ │ │ │ └── test_preprocess_data.sh │ │ │ │ │ │ └── vit_dataset.py │ │ │ │ │ ├── enums.py │ │ │ │ │ ├── fp16_deprecated │ │ │ │ │ │ └── loss_scaler.py │ │ │ │ │ ├── fused_kernels │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── compat.h │ │ │ │ │ │ ├── layer_norm_cuda.cpp │ │ │ │ │ │ ├── layer_norm_cuda_kernel.cu │ │ │ │ │ │ ├── scaled_masked_softmax.cpp │ │ │ │ │ │ ├── scaled_masked_softmax.h │ │ │ │ │ │ ├── scaled_masked_softmax_cuda.cu │ │ │ │ │ │ ├── scaled_upper_triang_masked_softmax.cpp │ │ │ │ │ │ ├── scaled_upper_triang_masked_softmax.h │ │ │ │ │ │ ├── scaled_upper_triang_masked_softmax_cuda.cu │ │ │ │ │ │ └── type_shim.h │ │ │ │ │ ├── global_vars.py │ │ │ │ │ ├── indexer.py │ │ │ │ │ ├── initialize.py │ │ │ │ │ ├── learning_rates.py │ │ │ │ │ ├── memory.py │ │ │ │ │ ├── microbatches.py │ │ │ │ │ ├── model │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── bert_model.py │ │ │ │ │ │ ├── biencoder_model.py │ │ │ │ │ │ ├── classification.py │ │ │ │ │ │ ├── distributed.py │ │ │ │ │ │ ├── enums.py │ │ │ │ │ │ ├── fused_bias_gelu.py │ │ │ │ │ │ ├── fused_layer_norm.py │ │ │ │ │ │ ├── fused_softmax.py │ │ │ │ │ │ ├── gpt_model.py │ │ │ │ │ │ ├── language_model.py │ │ │ │ │ │ ├── module.py │ │ │ │ │ │ ├── multiple_choice.py │ │ │ │ │ │ ├── realm_model.py │ │ │ │ │ │ ├── t5_model.py │ │ │ │ │ │ ├── transformer.py │ │ │ │ │ │ ├── utils.py │ │ │ │ │ │ └── vit_model.py │ │ │ │ │ ├── mpu │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── cross_entropy.py │ │ │ │ │ │ ├── data.py │ │ │ │ │ │ ├── initialize.py │ │ │ │ │ │ ├── layers.py │ │ │ │ │ │ ├── mappings.py │ │ │ │ │ │ ├── random.py │ │ │ │ │ │ ├── tests │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── commons.py │ │ │ │ │ │ │ ├── test_cross_entropy.py │ │ │ │ │ │ │ ├── test_data.py │ │ │ │ │ │ │ ├── test_initialize.py │ │ │ │ │ │ │ ├── test_layers.py │ │ │ │ │ │ │ └── test_random.py │ │ │ │ │ │ └── utils.py │ │ │ │ │ ├── optimizer │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── clip_grads.py │ │ │ │ │ │ ├── grad_scaler.py │ │ │ │ │ │ └── optimizer.py │ │ │ │ │ ├── p2p_communication.py │ │ │ │ │ ├── package_info.py │ │ │ │ │ ├── schedules.py │ │ │ │ │ ├── text_generation_utils.py │ │ │ │ │ ├── tokenizer │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── bert_tokenization.py │ │ │ │ │ │ ├── gpt2_tokenization.py │ │ │ │ │ │ └── tokenizer.py │ │ │ │ │ ├── training.py │ │ │ │ │ └── utils.py │ │ │ │ │ ├── patches │ │ │ │ │ ├── functional_changes.diff │ │ │ │ │ └── performance_patch_1.diff │ │ │ │ │ ├── pretrain_bert.py │ │ │ │ │ ├── pretrain_gpt.py │ │ │ │ │ ├── pretrain_ict.py │ │ │ │ │ ├── pretrain_t5.py │ │ │ │ │ ├── pretrain_vit.py │ │ │ │ │ ├── requirements.txt │ │ │ │ │ ├── scripts │ │ │ │ │ └── run_bloom13b.sh │ │ │ │ │ ├── setup.py │ │ │ │ │ ├── tasks │ │ │ │ │ ├── data_utils.py │ │ │ │ │ ├── ensemble_classifier.py │ │ │ │ │ ├── eval_harness │ │ │ │ │ │ ├── download.py │ │ │ │ │ │ ├── evaluate.py │ │ │ │ │ │ └── report-to-csv.py │ │ │ │ │ ├── eval_utils.py │ │ │ │ │ ├── finetune_utils.py │ │ │ │ │ ├── glue │ │ │ │ │ │ ├── data.py │ │ │ │ │ │ ├── finetune.py │ │ │ │ │ │ ├── mnli.py │ │ │ │ │ │ └── qqp.py │ │ │ │ │ ├── main.py │ │ │ │ │ ├── orqa │ │ │ │ │ │ ├── evaluate_orqa.py │ │ │ │ │ │ ├── evaluate_utils.py │ │ │ │ │ │ └── natural_questions │ │ │ │ │ │ │ ├── nq.py │ │ │ │ │ │ │ ├── qa_utils.py │ │ │ │ │ │ │ └── tokenizers.py │ │ │ │ │ ├── race │ │ │ │ │ │ ├── data.py │ │ │ │ │ │ └── finetune.py │ │ │ │ │ ├── vision │ │ │ │ │ │ ├── classification.py │ │ │ │ │ │ ├── eval_utils.py │ │ │ │ │ │ ├── finetune_utils.py │ │ │ │ │ │ └── main.py │ │ │ │ │ └── zeroshot_gpt │ │ │ │ │ │ ├── datasets.py │ │ │ │ │ │ ├── detokenizer.py │ │ │ │ │ │ └── evaluate.py │ │ │ │ │ ├── tests │ │ │ │ │ └── test_basic.py │ │ │ │ │ └── tools │ │ │ │ │ ├── convert_checkpoint │ │ │ │ │ ├── README.md │ │ │ │ │ ├── deepspeed_checkpoint.py │ │ │ │ │ ├── deepspeed_to_megatron.py │ │ │ │ │ ├── deepspeed_to_transformers.py │ │ │ │ │ ├── inspect_checkpoint.py │ │ │ │ │ └── inspect_deepspeed_checkpoint.py │ │ │ │ │ ├── create_doc_index.py │ │ │ │ │ ├── generate_samples_gpt.py │ │ │ │ │ ├── linter.py │ │ │ │ │ ├── merge_mp_partitions.py │ │ │ │ │ ├── openwebtext │ │ │ │ │ ├── README.md │ │ │ │ │ ├── add_id.py │ │ │ │ │ ├── blacklist_urls.py │ │ │ │ │ ├── cleanup_dataset.py │ │ │ │ │ ├── cleanup_fix_dataset.py │ │ │ │ │ ├── filter_ngrams.py │ │ │ │ │ ├── find_duplicates.py │ │ │ │ │ ├── group_duplicate_url.py │ │ │ │ │ ├── merge_jsons.py │ │ │ │ │ └── remove_group_duplicates.py │ │ │ │ │ └── preprocess_data.py │ │ │ └── bert │ │ │ │ ├── Dockerfile │ │ │ │ ├── LICENSE │ │ │ │ ├── NOTICE │ │ │ │ ├── README.md │ │ │ │ ├── bert_config.json │ │ │ │ ├── bind.sh │ │ │ │ ├── bind_pyt.py │ │ │ │ ├── checkpoints │ │ │ │ └── .keep │ │ │ │ ├── configurations.yml │ │ │ │ ├── create_pretraining_data.py │ │ │ │ ├── data │ │ │ │ ├── BooksDownloader.py │ │ │ │ ├── BookscorpusTextFormatting.py │ │ │ │ ├── Downloader.py │ │ │ │ ├── GLUEDownloader.py │ │ │ │ ├── GooglePretrainedWeightDownloader.py │ │ │ │ ├── NVIDIAPretrainedWeightDownloader.py │ │ │ │ ├── SquadDownloader.py │ │ │ │ ├── TextSharding.py │ │ │ │ ├── WikiDownloader.py │ │ │ │ ├── WikicorpusTextFormatting.py │ │ │ │ ├── __init__.py │ │ │ │ ├── bertPrep.py │ │ │ │ ├── create_datasets_from_start.sh │ │ │ │ └── squad │ │ │ │ │ └── squad_download.sh │ │ │ │ ├── extract_features.py │ │ │ │ ├── file_utils.py │ │ │ │ ├── gpu_migration_logs │ │ │ │ └── gpu_migration_5494.log │ │ │ │ ├── inference.py │ │ │ │ ├── modeling.py │ │ │ │ ├── optimization.py │ │ │ │ ├── patches │ │ │ │ ├── minimal_changes.diff │ │ │ │ ├── performance_improvements.diff │ │ │ │ └── use_packed_dataset.diff │ │ │ │ ├── processors │ │ │ │ ├── __init__.py │ │ │ │ └── glue.py │ │ │ │ ├── requirements.txt │ │ │ │ ├── results │ │ │ │ ├── .keep │ │ │ │ ├── checkpoints │ │ │ │ │ └── lddl_log │ │ │ │ │ │ ├── node-0.txt │ │ │ │ │ │ └── node-0_local-0.txt │ │ │ │ └── dllogger.json │ │ │ │ ├── run.sub │ │ │ │ ├── run_glue.py │ │ │ │ ├── run_pretraining.py │ │ │ │ ├── run_squad.py │ │ │ │ ├── run_swag.py │ │ │ │ ├── schedulers.py │ │ │ │ ├── scripts │ │ │ │ ├── configs │ │ │ │ │ ├── glue_config.sh │ │ │ │ │ ├── pretrain_config.sh │ │ │ │ │ └── squad_config.sh │ │ │ │ ├── data_download.sh │ │ │ │ ├── docker │ │ │ │ │ ├── build.sh │ │ │ │ │ └── launch.sh │ │ │ │ ├── run_glue.sh │ │ │ │ ├── run_pretraining.sh │ │ │ │ ├── run_squad.sh │ │ │ │ └── run_swag.sh │ │ │ │ ├── tokenization.py │ │ │ │ ├── utils.py │ │ │ │ └── vocab │ │ │ │ └── vocab │ │ └── simple_examples │ │ │ └── mnist │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── gpu_migration_logs │ │ │ └── gpu_migration_66.log │ │ │ └── main.py │ └── multi_tenants │ │ ├── README.md │ │ └── multi_tenants_resnet_pt.sh ├── generative_models │ └── stable-diffusion │ │ ├── LICENSE │ │ ├── README.md │ │ ├── configs │ │ ├── autoencoder │ │ │ ├── autoencoder_kl_16x16x16.yaml │ │ │ ├── autoencoder_kl_32x32x4.yaml │ │ │ ├── autoencoder_kl_64x64x3.yaml │ │ │ └── autoencoder_kl_8x8x64.yaml │ │ ├── latent-diffusion │ │ │ ├── celebahq-ldm-vq-4.yaml │ │ │ ├── cin-ldm-vq-f8.yaml │ │ │ ├── cin256-v2.yaml │ │ │ ├── ffhq-ldm-vq-4.yaml │ │ │ ├── lsun_bedrooms-ldm-vq-4.yaml │ │ │ ├── lsun_churches-ldm-kl-8.yaml │ │ │ └── txt2img-1p4B-eval.yaml │ │ └── stable-diffusion │ │ │ ├── dev.yaml │ │ │ ├── dev_mn.yaml │ │ │ ├── dev_mn_dummy.yaml │ │ │ ├── inpainting │ │ │ ├── v1-edgeinpainting.yaml │ │ │ ├── v1-finetune-for-inpainting-laion-aesthetic-larger-masks-and-ucfg.yaml │ │ │ ├── v1-finetune-for-inpainting-laion-aesthetic-larger-masks.yaml │ │ │ └── v1-finetune-for-inpainting-laion-iaesthe.yaml │ │ │ ├── txt2img-1p4B-multinode-clip-encoder-high-res-512.yaml │ │ │ ├── txt2img-1p4B-multinode-clip-encoder.yaml │ │ │ ├── txt2img-1p4B-multinode-t5-encoder.yaml │ │ │ ├── txt2img-1p4B-multinode.yaml │ │ │ ├── txt2img-clip-encoder-dev.yaml │ │ │ ├── txt2img-ldm-frozen-dev.yaml │ │ │ ├── txt2img-ldm-unfrozen-dev.yaml │ │ │ ├── txt2img-ldm-vae-f8.yaml │ │ │ ├── txt2img-multinode-clip-encoder-f16-1024-laion-hr.yaml │ │ │ ├── txt2img-multinode-clip-encoder-f16-256-pretraining.yaml │ │ │ ├── txt2img-multinode-clip-encoder-f16-768-laion-hr-inference.yaml │ │ │ ├── txt2img-multinode-clip-encoder-f16-768-laion-hr.yaml │ │ │ ├── txt2img-multinode-clip-encoder-f16-768.yaml │ │ │ ├── txt2img-t5-encoder-dev.yaml │ │ │ ├── txt2img-upscale-clip-encoder-f16-1024-dev.yaml │ │ │ ├── txt2img-upscale-clip-encoder-f16-1024.yaml │ │ │ ├── txt2img-v2-clip-encoder-improved_aesthetics-256-dev.yaml │ │ │ ├── txt2img-v2-clip-encoder-improved_aesthetics-256.yaml │ │ │ ├── txt2img-v2-clip-encoder-improved_aesthetics-512-dev.yaml │ │ │ ├── txt2img-v2-clip-encoder-improved_aesthetics-512.yaml │ │ │ ├── upscaling │ │ │ └── upscale-v1-with-f16.yaml │ │ │ ├── v1-inference.yaml │ │ │ ├── v1_improvedaesthetics.yaml │ │ │ ├── v1_laionhr.yaml │ │ │ ├── v2_laionhr1024.yaml │ │ │ ├── v2_laionhr1024_2.yaml │ │ │ ├── v2_pretraining.yaml │ │ │ └── v3_pretraining.yaml │ │ ├── cpu_config.yaml │ │ ├── environment.yaml │ │ ├── hpu_config_web_dataset.yaml │ │ ├── ldm │ │ ├── data │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── coco.py │ │ │ ├── dummy.py │ │ │ ├── imagenet.py │ │ │ ├── inpainting │ │ │ │ ├── __init__.py │ │ │ │ └── synthetic_mask.py │ │ │ ├── laion.py │ │ │ └── lsun.py │ │ ├── lr_scheduler.py │ │ ├── models │ │ │ ├── autoencoder.py │ │ │ └── diffusion │ │ │ │ ├── __init__.py │ │ │ │ ├── classifier.py │ │ │ │ ├── ddim.py │ │ │ │ ├── ddpm.py │ │ │ │ ├── plms.py │ │ │ │ └── sampling_util.py │ │ ├── modules │ │ │ ├── attention.py │ │ │ ├── diffusionmodules │ │ │ │ ├── __init__.py │ │ │ │ ├── model.py │ │ │ │ ├── openaimodel.py │ │ │ │ └── util.py │ │ │ ├── distributions │ │ │ │ ├── __init__.py │ │ │ │ └── distributions.py │ │ │ ├── ema.py │ │ │ ├── encoders │ │ │ │ ├── __init__.py │ │ │ │ └── modules.py │ │ │ ├── evaluate │ │ │ │ ├── adm_evaluator.py │ │ │ │ ├── evaluate_perceptualsim.py │ │ │ │ ├── frechet_video_distance.py │ │ │ │ ├── ssim.py │ │ │ │ └── torch_frechet_video_distance.py │ │ │ ├── image_degradation │ │ │ │ ├── __init__.py │ │ │ │ ├── bsrgan.py │ │ │ │ ├── bsrgan_light.py │ │ │ │ ├── utils │ │ │ │ │ └── test.png │ │ │ │ └── utils_image.py │ │ │ ├── losses │ │ │ │ ├── __init__.py │ │ │ │ ├── contperceptual.py │ │ │ │ └── vqperceptual.py │ │ │ └── x_transformer.py │ │ └── util.py │ │ ├── main.py │ │ ├── models │ │ ├── first_stage_models │ │ │ ├── kl-f16 │ │ │ │ └── config.yaml │ │ │ ├── kl-f32 │ │ │ │ └── config.yaml │ │ │ ├── kl-f4 │ │ │ │ └── config.yaml │ │ │ ├── kl-f8 │ │ │ │ └── config.yaml │ │ │ ├── vq-f16 │ │ │ │ └── config.yaml │ │ │ ├── vq-f4-noattn │ │ │ │ └── config.yaml │ │ │ ├── vq-f4 │ │ │ │ └── config.yaml │ │ │ ├── vq-f8-n256 │ │ │ │ └── config.yaml │ │ │ └── vq-f8 │ │ │ │ └── config.yaml │ │ └── ldm │ │ │ ├── bsr_sr │ │ │ └── config.yaml │ │ │ ├── celeba256 │ │ │ └── config.yaml │ │ │ ├── cin256 │ │ │ └── config.yaml │ │ │ ├── ffhq256 │ │ │ └── config.yaml │ │ │ ├── inpainting_big │ │ │ └── config.yaml │ │ │ ├── layout2img-openimages256 │ │ │ └── config.yaml │ │ │ ├── lsun_beds256 │ │ │ └── config.yaml │ │ │ ├── lsun_churches256 │ │ │ └── config.yaml │ │ │ ├── semantic_synthesis256 │ │ │ └── config.yaml │ │ │ ├── semantic_synthesis512 │ │ │ └── config.yaml │ │ │ └── text2img256 │ │ │ └── config.yaml │ │ ├── notebook_helpers.py │ │ ├── ops_bf16.txt │ │ ├── ops_fp32.txt │ │ ├── requirements.txt │ │ ├── scripts │ │ ├── autoencoder-eval.py │ │ ├── checker.py │ │ ├── cmd_on_new_ckpt.py │ │ ├── demo │ │ │ └── inpainting.py │ │ ├── img2img.py │ │ ├── inpaint.py │ │ ├── inpaint_sd.py │ │ ├── logging_template.py │ │ ├── mnist-distributed.py │ │ ├── printckpt.py │ │ ├── prompts │ │ │ ├── aesthetic-prompts-plain.txt │ │ │ ├── aesthetic-prompts-surrealism.txt │ │ │ ├── prompts-with-wings.txt │ │ │ ├── six-prompts │ │ │ ├── weird-dalle-prompts.txt │ │ │ ├── wings1.txt │ │ │ ├── wings2.txt │ │ │ ├── wings3.txt │ │ │ └── wings4.txt │ │ ├── prune-ckpt.py │ │ ├── sample_diffusion.py │ │ ├── slurm │ │ │ ├── README.md │ │ │ ├── eval_inpainting │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── resume_512 │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── resume_512_improvedaesthetic │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── resume_768_hr │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v1-upscaling-f16-pretraining-512-aesthetics │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v1_edgeinpainting │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v1_iahr_torch111 │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v1_iahr_torch111_ucg │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v1_improvedaesthetics │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v1_improvedaesthetics_torch111 │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v1_inpainting_aesthetics-larger-masks-ucg │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v1_inpainting_aesthetics-larger-masks │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v1_inpainting_improvedaesthetics_torch111 │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v1_laionhr_torch111 │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v2_laionhr1024 │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v2_laionhr1024_2 │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ ├── v2_pretraining │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ │ └── v3_pretraining │ │ │ │ ├── launcher.sh │ │ │ │ └── sbatch.sh │ │ ├── test_gpu.py │ │ ├── test_gpu.sh │ │ ├── txt2img.py │ │ └── vqgan_codebook_visualizer.py │ │ └── setup.py └── nlp │ ├── DeepSpeedExamples │ └── deepspeed-bert │ │ ├── LICENSE │ │ ├── README.md │ │ ├── create_pretraining_data.py │ │ ├── data │ │ ├── BooksDownloader.py │ │ ├── BookscorpusTextFormatting.py │ │ ├── Downloader.py │ │ ├── GooglePretrainedWeightDownloader.py │ │ ├── TextSharding.py │ │ ├── WikiDownloader.py │ │ ├── WikicorpusTextFormatting.py │ │ ├── __init__.py │ │ ├── bertPrep.py │ │ └── create_datasets_from_start.sh │ │ ├── file_utils.py │ │ ├── lamb.py │ │ ├── lamb_exp.py │ │ ├── lans.py │ │ ├── modeling.py │ │ ├── requirements.txt │ │ ├── run_pretraining.py │ │ ├── schedulers.py │ │ ├── scripts │ │ ├── bert_1.5b_config.json │ │ ├── bert_5b_config.json │ │ ├── deepspeed_config_bert_1.5b.json │ │ ├── deepspeed_config_bert_5b_lans.json │ │ ├── hostsfile │ │ ├── run_bert_1.5b_8x.sh │ │ └── run_bert_5b_32x_lans.sh │ │ ├── tokenization.py │ │ └── utils.py │ └── bert │ ├── LICENSE │ ├── README.md │ ├── bert_config.json │ ├── bert_config_1.2B.json │ ├── create_pretraining_data.py │ ├── data │ ├── BooksDownloader.py │ ├── BookscorpusTextFormatting.py │ ├── Downloader.py │ ├── GooglePretrainedWeightDownloader.py │ ├── TextSharding.py │ ├── WikiDownloader.py │ ├── WikicorpusTextFormatting.py │ ├── __init__.py │ ├── bertPrep.py │ ├── create_datasets_from_start.sh │ └── squad │ │ └── squad_download.sh │ ├── file_utils.py │ ├── lamb.py │ ├── modeling.py │ ├── optimization.py │ ├── pack_pretraining_data_pytorch.py │ ├── pytorch_packed_data_checker.py │ ├── requirements.txt │ ├── run_pretraining.py │ ├── run_squad.py │ ├── schedulers.py │ ├── scripts │ └── run_pretraining.sh │ ├── tokenization.py │ └── utils.py ├── README.md ├── ethics.md └── security.md /.github/CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/.github/CODEOWNERS -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /PyTorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/LICENSE -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/data_loaders.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/googlenet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/googlenet_utils.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/inference.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/main.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/media_pipe_settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/media_pipe_settings.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/model/__init__.py: -------------------------------------------------------------------------------- 1 | from .resnet import * 2 | -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/model/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/model/optimizer.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/model/resnet.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/model/utils.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/ops_bf16_Resnet.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/ops_bf16_Resnet.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/ops_fp32_Resnet.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/ops_fp32_Resnet.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/requirements.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/requirements_u24.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/requirements_u24.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/resnet_media_pipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/resnet_media_pipe.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/train.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/classification/torchvision/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/classification/torchvision/utils.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/LICENSE -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/assets/demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/assets/demo.png -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/assets/dog.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/assets/dog.jpg -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/assets/git_fig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/assets/git_fig.png -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/assets/logo.png -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/cpp/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/cpp/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/cpp/build.sh -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/cpp/yolox.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/cpp/yolox.cpp -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/build.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/convert_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/convert_weights.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/demo.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/dump.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/dump.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/darknet.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/network_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/network_blocks.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/yolo_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/yolo_fpn.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/yolo_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/yolo_head.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/yolo_pafpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/yolo_pafpn.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/yolox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/MegEngine/python/models/yolox.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ONNXRuntime/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ONNXRuntime/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ONNXRuntime/onnx_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ONNXRuntime/onnx_inference.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/OpenVINO/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/OpenVINO/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/OpenVINO/cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/OpenVINO/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/OpenVINO/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/OpenVINO/cpp/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/OpenVINO/cpp/yolox_openvino.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/OpenVINO/cpp/yolox_openvino.cpp -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/OpenVINO/python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/OpenVINO/python/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/OpenVINO/python/openvino_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/OpenVINO/python/openvino_inference.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/TensorRT/cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/TensorRT/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/TensorRT/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/TensorRT/cpp/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/TensorRT/cpp/logging.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/TensorRT/cpp/logging.h -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/TensorRT/cpp/yolox.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/TensorRT/cpp/yolox.cpp -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/TensorRT/python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/TensorRT/python/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/build.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/build.gradle -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/src/main/AndroidManifest.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/src/main/AndroidManifest.xml -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/src/main/assets/yolox.param: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/src/main/assets/yolox.param -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/src/main/jni/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/src/main/jni/CMakeLists.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/src/main/jni/yoloXncnn_jni.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/src/main/jni/yoloXncnn_jni.cpp -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/src/main/res/layout/main.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/src/main/res/layout/main.xml -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/src/main/res/values/strings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/app/src/main/res/values/strings.xml -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/build.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/build.gradle -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/gradle/wrapper/gradle-wrapper.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/gradle/wrapper/gradle-wrapper.jar -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/gradle/wrapper/gradle-wrapper.properties: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/gradle/wrapper/gradle-wrapper.properties -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/gradlew: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/gradlew -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/gradlew.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/android/gradlew.bat -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/android/settings.gradle: -------------------------------------------------------------------------------- 1 | include ':app' 2 | -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/cpp/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/demo/ncnn/cpp/yolox.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/demo/ncnn/cpp/yolox.cpp -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/.gitignore: -------------------------------------------------------------------------------- 1 | _build -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/docs/Makefile -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/_static/css/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/docs/_static/css/custom.css -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/docs/conf.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/demo/megengine_cpp_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/MegEngine/cpp/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/demo/megengine_py_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/MegEngine/python/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/demo/ncnn_android_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/ncnn/android/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/demo/ncnn_cpp_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/ncnn/cpp/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/demo/onnx_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/ONNXRuntime/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/demo/openvino_cpp_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/OpenVINO/cpp/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/demo/openvino_py_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/OpenVINO/python/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/demo/trt_cpp_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/TensorRT/cpp/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/demo/trt_py_readme.md: -------------------------------------------------------------------------------- 1 | ../../demo/TensorRT/python/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/docs/index.rst -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/manipulate_training_image_size.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/docs/manipulate_training_image_size.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/model_zoo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/docs/model_zoo.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/quick_run.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/docs/quick_run.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/requirements-doc.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/docs/requirements-doc.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/train_custom_data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/docs/train_custom_data.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/docs/updates_note.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/docs/updates_note.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/download_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/download_dataset.sh -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/exps/default/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/exps/default/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/exps/default/yolov3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/exps/default/yolov3.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/exps/default/yolox_l.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/exps/default/yolox_l.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/exps/default/yolox_m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/exps/default/yolox_m.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/exps/default/yolox_nano.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/exps/default/yolox_nano.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/exps/default/yolox_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/exps/default/yolox_s.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/exps/default/yolox_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/exps/default/yolox_tiny.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/exps/default/yolox_x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/exps/default/yolox_x.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/exps/example/custom/nano.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/exps/example/custom/nano.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/exps/example/custom/yolox_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/exps/example/custom/yolox_s.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/exps/example/yolox_voc/yolox_voc_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/exps/example/yolox_voc/yolox_voc_s.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/hubconf.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/ops_bf16_yolox.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/ops_bf16_yolox.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/ops_fp32_yolox.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/ops_fp32_yolox.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/requirements.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/setup.cfg -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/setup.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/tests/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python3 2 | # -*- coding:utf-8 -*- 3 | -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/tests/utils/test_model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/tests/utils/test_model_utils.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/tools/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/tools/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/tools/demo.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/tools/eval.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/tools/export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/tools/export_onnx.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/tools/export_torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/tools/export_torchscript.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/tools/train.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/tools/trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/tools/trt.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/core/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/core/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/core/launch.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/core/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/core/trainer.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/data_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/data_augment.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/data_prefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/data_prefetcher.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/dataloading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/dataloading.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/dataloading_hpu_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/dataloading_hpu_loader.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/dataloading_hpu_mediapipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/dataloading_hpu_mediapipe.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/datasets/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/datasets/coco.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/datasets/coco_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/datasets/coco_classes.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/datasets/datasets_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/datasets/datasets_wrapper.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/datasets/mosaicdetection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/datasets/mosaicdetection.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/datasets/voc.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/datasets/voc_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/datasets/voc_classes.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/data/samplers.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/evaluators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/evaluators/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/evaluators/coco_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/evaluators/coco_evaluator.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/evaluators/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/evaluators/voc_eval.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/evaluators/voc_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/evaluators/voc_evaluator.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/exp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/exp/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/exp/base_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/exp/base_exp.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/exp/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/exp/build.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/exp/default/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/exp/default/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/exp/yolox_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/exp/yolox_base.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/layers/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/layers/csrc/cocoeval/cocoeval.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/layers/csrc/cocoeval/cocoeval.cpp -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/layers/csrc/cocoeval/cocoeval.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/layers/csrc/cocoeval/cocoeval.h -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/layers/csrc/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/layers/csrc/vision.cpp -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/layers/fast_coco_eval_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/layers/fast_coco_eval_api.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/models/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/models/build.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/models/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/models/darknet.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/models/losses.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/models/network_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/models/network_blocks.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/models/yolo_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/models/yolo_fpn.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/models/yolo_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/models/yolo_head.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/models/yolo_head_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/models/yolo_head_script.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/models/yolo_pafpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/models/yolo_pafpn.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/models/yolox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/models/yolox.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/tools/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/__init__.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/allreduce_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/allreduce_norm.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/boxes.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/checkpoint.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/compat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/compat.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/demo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/demo_utils.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/dist.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/ema.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/logger.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/lr_scheduler.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/metric.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/model_utils.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/setup_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/setup_env.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/detection/yolox/yolox/utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/detection/yolox/yolox/utils/visualize.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/Dockerfile -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/LICENSE -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/README.md -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/config/ops_bf16_unet.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/config/ops_bf16_unet.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/config/ops_fp32_unet.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/config/ops_fp32_unet.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/data_loading/dali_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/data_loading/dali_loader.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/data_loading/data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/data_loading/data_module.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/data_preprocessing/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/data_preprocessing/configs.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/data_preprocessing/preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/data_preprocessing/preprocessor.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/download.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/evaluate.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/images/unet3d.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/images/unet3d.png -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/lightning_trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/lightning_trainer/ptl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/lightning_trainer/ptl.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/main.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/models/dice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/models/dice.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/models/layers.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/models/loss.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/models/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/models/metrics.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/models/monai_sliding_window_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/models/monai_sliding_window_inference.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/models/nn_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/models/nn_unet.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/models/pl_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/models/pl_metric.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/models/unet.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/preprocess.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/pytorch/early_stopping_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/pytorch/early_stopping_unet.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/pytorch/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/pytorch/misc.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/pytorch/npt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/pytorch/npt.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/pytorch/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/pytorch/trainer.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/requirements.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/requirements_u22.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/requirements_u22.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/requirements_u24.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/requirements_u24.txt -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/scripts/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/scripts/benchmark.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/scripts/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/scripts/inference.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/scripts/train.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/utils/early_stopping_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/utils/early_stopping_unet.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/utils/gpu_affinity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/utils/gpu_affinity.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/utils/logger.py -------------------------------------------------------------------------------- /PyTorch/computer_vision/segmentation/Unet/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/computer_vision/segmentation/Unet/utils/utils.py -------------------------------------------------------------------------------- /PyTorch/examples/DeepSpeed/cifar_example/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/DeepSpeed/cifar_example/LICENSE -------------------------------------------------------------------------------- /PyTorch/examples/DeepSpeed/cifar_example/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/DeepSpeed/cifar_example/README.md -------------------------------------------------------------------------------- /PyTorch/examples/DeepSpeed/cifar_example/cifar10_deepspeed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/DeepSpeed/cifar_example/cifar10_deepspeed.py -------------------------------------------------------------------------------- /PyTorch/examples/DeepSpeed/cifar_example/ds_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/DeepSpeed/cifar_example/ds_config.json -------------------------------------------------------------------------------- /PyTorch/examples/DeepSpeed/cifar_example/requirements.txt: -------------------------------------------------------------------------------- 1 | pillow>=7.1.0 2 | matplotlib 3 | torchmetrics>=0.8.0 4 | -------------------------------------------------------------------------------- /PyTorch/examples/DeepSpeed/cifar_example/run_ds_habanax8.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/DeepSpeed/cifar_example/run_ds_habanax8.sh -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/README.md -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/brute_force_min_pad_waste.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/brute_force_min_pad_waste.py -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/bucket.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/bucket.py -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/bucket_analysis.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/bucket_analysis.svg -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/bucket_analysis_bar_gaussian.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/bucket_analysis_bar_gaussian.svg -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/bucket_analysis_bar_squad.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/bucket_analysis_bar_squad.svg -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/bucket_analysis_num_steps_gaussian.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/bucket_analysis_num_steps_gaussian.svg -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/datasets_library.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/datasets_library.py -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/gaussian.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/gaussian.svg -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/lloyd_max_bucket.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/lloyd_max_bucket.py -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/plotting.py -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/requirements.txt -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/run_demo_bucketing_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/run_demo_bucketing_gaussian.py -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/run_demo_bucketing_squad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/run_demo_bucketing_squad.py -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/run_demo_controlling_num_steps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/run_demo_controlling_num_steps.py -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/run_demo_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/run_demo_gaussian.py -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/run_demo_squad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/run_demo_squad.py -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/squad.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/squad.svg -------------------------------------------------------------------------------- /PyTorch/examples/bucketing/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/bucketing/test.py -------------------------------------------------------------------------------- /PyTorch/examples/computer_vision/hello_world/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/computer_vision/hello_world/README.md -------------------------------------------------------------------------------- /PyTorch/examples/computer_vision/hello_world/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/computer_vision/hello_world/example.py -------------------------------------------------------------------------------- /PyTorch/examples/computer_vision/hello_world/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/computer_vision/hello_world/mnist.py -------------------------------------------------------------------------------- /PyTorch/examples/computer_vision/hello_world/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/computer_vision/hello_world/utils.py -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/custom_fusedsdpa/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/custom_fusedsdpa/README.md -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/custom_fusedsdpa/custom_fusedsdpa_op.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/custom_fusedsdpa/custom_fusedsdpa_op.patch -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/README.md -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/__init__.py -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/custom_relu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/custom_relu.py -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/custom_relu_op.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/custom_relu_op.patch -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/hpu_custom_op_relu_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/hpu_custom_op_relu_test.py -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/hpu_custom_relu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/hpu_custom_relu.cpp -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/legacy_custom_op_API/custom_relu/setup.py -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/legacy_custom_op_API/custom_topk/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/legacy_custom_op_API/custom_topk/README.md -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/legacy_custom_op_API/custom_topk/hpu_custom_op_topk_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/legacy_custom_op_API/custom_topk/hpu_custom_op_topk_test.py -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/legacy_custom_op_API/custom_topk/hpu_custom_topk.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/legacy_custom_op_API/custom_topk/hpu_custom_topk.cpp -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/legacy_custom_op_API/custom_topk/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/legacy_custom_op_API/custom_topk/setup.py -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/README.md -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/__init__.py -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/custom_relu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/custom_relu.py -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/custom_relu_op.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/custom_relu_op.patch -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/hpu_custom_op_relu_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/hpu_custom_op_relu_test.py -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/hpu_custom_relu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/hpu_custom_relu.cpp -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/pt2_custom_op_API/custom_relu/setup.py -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/pt2_custom_op_API/custom_topk/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/pt2_custom_op_API/custom_topk/README.md -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/pt2_custom_op_API/custom_topk/hpu_custom_topk.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/pt2_custom_op_API/custom_topk/hpu_custom_topk.cpp -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/pt2_custom_op_API/custom_topk/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/pt2_custom_op_API/custom_topk/setup.py -------------------------------------------------------------------------------- /PyTorch/examples/custom_op/pt2_custom_op_API/custom_topk/test_hpu_custom_op_topk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/custom_op/pt2_custom_op_API/custom_topk/test_hpu_custom_op_topk.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/README.md -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/LICENSE -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/README.md -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/presets.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/sampler.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/train.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/train_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/train_quantization.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/transforms.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/computer_vision/classification/torchvision/utils.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/LICENSE -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/README.md -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/environment.yaml -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/hpu_graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/hpu_graph_utils.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/base.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/coco.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/dummy.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/imagenet.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/inpainting/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/laion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/laion.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/data/lsun.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/lr_scheduler.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/attention.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/ema.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/evaluate/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/evaluate/ssim.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from ldm.modules.losses.contperceptual import LPIPSWithDiscriminator -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/modules/x_transformer.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/ldm/util.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/main.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/models/ldm/bsr_sr/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/models/ldm/bsr_sr/config.yaml -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/models/ldm/celeba256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/models/ldm/celeba256/config.yaml -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/models/ldm/cin256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/models/ldm/cin256/config.yaml -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/models/ldm/ffhq256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/models/ldm/ffhq256/config.yaml -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/notebook_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/notebook_helpers.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/patches/hpu_graph.diff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/patches/hpu_graph.diff -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/patches/minimal_changes.diff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/patches/minimal_changes.diff -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/patches/randn_to_cpu.diff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/patches/randn_to_cpu.diff -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/requirements.txt -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/autoencoder-eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/autoencoder-eval.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/checker.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/cmd_on_new_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/cmd_on_new_ckpt.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/demo/inpainting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/demo/inpainting.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/download_first_stages.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/download_first_stages.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/download_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/download_models.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/img2img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/img2img.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/inpaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/inpaint.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/logging_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/logging_template.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/mnist-distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/mnist-distributed.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/printckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/printckpt.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/prompts/six-prompts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/prompts/six-prompts -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/prompts/wings1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/prompts/wings1.txt -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/prompts/wings2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/prompts/wings2.txt -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/prompts/wings3.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/prompts/wings3.txt -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/prompts/wings4.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/prompts/wings4.txt -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/prune-ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/prune-ckpt.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/sample_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/sample_diffusion.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/slurm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/slurm/README.md -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/test_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/test_gpu.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/test_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/test_gpu.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/txt2img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/scripts/txt2img.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/generative_models/stable-diffusion/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/generative_models/stable-diffusion/setup.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/CODEOWNERS -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/LICENSE -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/MANIFEST.in -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/README.md -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/SECURITY.md -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/dataset/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/dataset/README.md -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/dataset/download_books.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/dataset/download_books.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/dataset/download_ckpt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/dataset/download_ckpt.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/dataset/download_vocab.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/dataset/download_vocab.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/README.md -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/azure/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/azure/README.md -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/azure/run-175b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/azure/run-175b.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/azure/run-1t.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/azure/run-1t.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/pretrain_gpt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/pretrain_gpt.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/pretrain_ict.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/pretrain_ict.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/pretrain_t5.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/examples/pretrain_t5.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/__init__.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/arguments.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/data/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/data/Makefile -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/data/__init__.py: -------------------------------------------------------------------------------- 1 | from . import indexed_dataset 2 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/enums.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/enums.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/global_vars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/global_vars.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/indexer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/indexer.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/initialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/initialize.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/memory.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/microbatches.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/microbatches.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/model/enums.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/model/enums.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/model/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/model/module.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/model/utils.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/__init__.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/data.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/layers.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/mappings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/mappings.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/random.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/mpu/utils.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/package_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/package_info.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/schedules.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/training.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/megatron/utils.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/pretrain_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/pretrain_bert.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/pretrain_gpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/pretrain_gpt.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/pretrain_ict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/pretrain_ict.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/pretrain_t5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/pretrain_t5.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/pretrain_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/pretrain_vit.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/requirements.txt: -------------------------------------------------------------------------------- 1 | pybind11 2 | torch 3 | six 4 | regex 5 | numpy 6 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/scripts/run_bloom13b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/scripts/run_bloom13b.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/setup.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/data_utils.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/eval_utils.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/finetune_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/finetune_utils.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/glue/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/glue/data.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/glue/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/glue/finetune.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/glue/mnli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/glue/mnli.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/glue/qqp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/glue/qqp.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/main.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/race/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/race/data.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/race/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/race/finetune.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/vision/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tasks/vision/main.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tests/test_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tests/test_basic.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tools/linter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tools/linter.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tools/preprocess_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/DeepSpeedExamples/Megatron-DeepSpeed/tools/preprocess_data.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/Dockerfile -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/LICENSE -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/NOTICE -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/README.md -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/bert_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/bert_config.json -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/bind.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/bind.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/bind_pyt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/bind_pyt.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/checkpoints/.keep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/configurations.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/configurations.yml -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/create_pretraining_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/create_pretraining_data.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/BooksDownloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/BooksDownloader.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/BookscorpusTextFormatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/BookscorpusTextFormatting.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/Downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/Downloader.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/GLUEDownloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/GLUEDownloader.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/GooglePretrainedWeightDownloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/GooglePretrainedWeightDownloader.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/NVIDIAPretrainedWeightDownloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/NVIDIAPretrainedWeightDownloader.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/SquadDownloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/SquadDownloader.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/TextSharding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/TextSharding.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/WikiDownloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/WikiDownloader.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/WikicorpusTextFormatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/WikicorpusTextFormatting.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/__init__.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/bertPrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/bertPrep.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/create_datasets_from_start.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/create_datasets_from_start.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/data/squad/squad_download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/data/squad/squad_download.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/extract_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/extract_features.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/file_utils.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/gpu_migration_logs/gpu_migration_5494.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/gpu_migration_logs/gpu_migration_5494.log -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/inference.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/modeling.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/optimization.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/patches/minimal_changes.diff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/patches/minimal_changes.diff -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/patches/performance_improvements.diff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/patches/performance_improvements.diff -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/patches/use_packed_dataset.diff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/patches/use_packed_dataset.diff -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/processors/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/processors/glue.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/processors/glue.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/requirements.txt -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/results/.keep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/results/checkpoints/lddl_log/node-0.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/results/checkpoints/lddl_log/node-0_local-0.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/results/dllogger.json: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/run.sub: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/run.sub -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/run_glue.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/run_glue.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/run_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/run_pretraining.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/run_squad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/run_squad.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/run_swag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/run_swag.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/schedulers.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/scripts/configs/glue_config.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/scripts/configs/glue_config.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/scripts/configs/pretrain_config.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/scripts/configs/pretrain_config.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/scripts/configs/squad_config.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/scripts/configs/squad_config.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/scripts/data_download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/scripts/data_download.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/scripts/docker/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/scripts/docker/build.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/scripts/docker/launch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/scripts/docker/launch.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/scripts/run_glue.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/scripts/run_glue.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/scripts/run_pretraining.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/scripts/run_pretraining.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/scripts/run_squad.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/scripts/run_squad.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/scripts/run_swag.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/scripts/run_swag.sh -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/tokenization.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/utils.py -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/nlp/bert/vocab/vocab: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/nlp/bert/vocab/vocab -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/simple_examples/mnist/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/simple_examples/mnist/LICENSE -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/simple_examples/mnist/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/simple_examples/mnist/README.md -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/simple_examples/mnist/gpu_migration_logs/gpu_migration_66.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/simple_examples/mnist/gpu_migration_logs/gpu_migration_66.log -------------------------------------------------------------------------------- /PyTorch/examples/gpu_migration/simple_examples/mnist/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/gpu_migration/simple_examples/mnist/main.py -------------------------------------------------------------------------------- /PyTorch/examples/multi_tenants/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/multi_tenants/README.md -------------------------------------------------------------------------------- /PyTorch/examples/multi_tenants/multi_tenants_resnet_pt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/examples/multi_tenants/multi_tenants_resnet_pt.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/LICENSE -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/README.md -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/autoencoder/autoencoder_kl_16x16x16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/autoencoder/autoencoder_kl_16x16x16.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/autoencoder/autoencoder_kl_32x32x4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/autoencoder/autoencoder_kl_32x32x4.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/autoencoder/autoencoder_kl_64x64x3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/autoencoder/autoencoder_kl_64x64x3.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/autoencoder/autoencoder_kl_8x8x64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/autoencoder/autoencoder_kl_8x8x64.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/celebahq-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/celebahq-ldm-vq-4.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/cin-ldm-vq-f8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/cin-ldm-vq-f8.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/cin256-v2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/cin256-v2.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/ffhq-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/ffhq-ldm-vq-4.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/lsun_bedrooms-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/lsun_bedrooms-ldm-vq-4.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/lsun_churches-ldm-kl-8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/lsun_churches-ldm-kl-8.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/txt2img-1p4B-eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/latent-diffusion/txt2img-1p4B-eval.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/dev.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/dev.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/dev_mn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/dev_mn.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/dev_mn_dummy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/dev_mn_dummy.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/txt2img-1p4B-multinode.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/txt2img-1p4B-multinode.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/txt2img-ldm-frozen-dev.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/txt2img-ldm-frozen-dev.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/txt2img-ldm-vae-f8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/txt2img-ldm-vae-f8.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/txt2img-t5-encoder-dev.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/txt2img-t5-encoder-dev.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v1-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v1-inference.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v1_improvedaesthetics.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v1_improvedaesthetics.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v1_laionhr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v1_laionhr.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v2_laionhr1024.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v2_laionhr1024.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v2_laionhr1024_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v2_laionhr1024_2.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v2_pretraining.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v2_pretraining.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v3_pretraining.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/configs/stable-diffusion/v3_pretraining.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/cpu_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/cpu_config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/environment.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/hpu_config_web_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/hpu_config_web_dataset.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/data/base.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/data/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/data/coco.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/data/dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/data/dummy.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/data/imagenet.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/data/inpainting/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/data/inpainting/synthetic_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/data/inpainting/synthetic_mask.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/data/laion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/data/laion.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/data/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/data/lsun.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/lr_scheduler.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/models/diffusion/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/models/diffusion/classifier.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/models/diffusion/sampling_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/models/diffusion/sampling_util.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/attention.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/ema.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/evaluate/adm_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/evaluate/adm_evaluator.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/evaluate/evaluate_perceptualsim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/evaluate/evaluate_perceptualsim.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/evaluate/frechet_video_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/evaluate/frechet_video_distance.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/evaluate/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/evaluate/ssim.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/evaluate/torch_frechet_video_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/evaluate/torch_frechet_video_distance.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/image_degradation/utils/test.png -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from ldm.modules.losses.contperceptual import LPIPSWithDiscriminator -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/losses/contperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/losses/contperceptual.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/modules/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/modules/x_transformer.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ldm/util.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/main.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/first_stage_models/kl-f16/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/first_stage_models/kl-f16/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/first_stage_models/kl-f32/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/first_stage_models/kl-f32/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/first_stage_models/kl-f4/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/first_stage_models/kl-f4/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/first_stage_models/kl-f8/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/first_stage_models/kl-f8/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/first_stage_models/vq-f16/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/first_stage_models/vq-f16/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/first_stage_models/vq-f4-noattn/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/first_stage_models/vq-f4-noattn/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/first_stage_models/vq-f4/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/first_stage_models/vq-f4/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/first_stage_models/vq-f8-n256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/first_stage_models/vq-f8-n256/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/first_stage_models/vq-f8/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/first_stage_models/vq-f8/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/ldm/bsr_sr/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/ldm/bsr_sr/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/ldm/celeba256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/ldm/celeba256/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/ldm/cin256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/ldm/cin256/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/ldm/ffhq256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/ldm/ffhq256/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/ldm/inpainting_big/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/ldm/inpainting_big/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/ldm/layout2img-openimages256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/ldm/layout2img-openimages256/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/ldm/lsun_beds256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/ldm/lsun_beds256/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/ldm/lsun_churches256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/ldm/lsun_churches256/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/ldm/semantic_synthesis256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/ldm/semantic_synthesis256/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/ldm/semantic_synthesis512/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/ldm/semantic_synthesis512/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/models/ldm/text2img256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/models/ldm/text2img256/config.yaml -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/notebook_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/notebook_helpers.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ops_bf16.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/ops_bf16.txt -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/ops_fp32.txt: -------------------------------------------------------------------------------- 1 | cross_entropy 2 | log_softmax 3 | nll_loss 4 | topk 5 | -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/requirements.txt -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/autoencoder-eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/autoencoder-eval.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/checker.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/cmd_on_new_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/cmd_on_new_ckpt.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/demo/inpainting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/demo/inpainting.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/img2img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/img2img.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/inpaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/inpaint.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/inpaint_sd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/inpaint_sd.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/logging_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/logging_template.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/mnist-distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/mnist-distributed.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/printckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/printckpt.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/prompts/aesthetic-prompts-plain.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/prompts/aesthetic-prompts-plain.txt -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/prompts/aesthetic-prompts-surrealism.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/prompts/aesthetic-prompts-surrealism.txt -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/prompts/prompts-with-wings.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/prompts/prompts-with-wings.txt -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/prompts/six-prompts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/prompts/six-prompts -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/prompts/weird-dalle-prompts.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/prompts/weird-dalle-prompts.txt -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/prompts/wings1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/prompts/wings1.txt -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/prompts/wings2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/prompts/wings2.txt -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/prompts/wings3.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/prompts/wings3.txt -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/prompts/wings4.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/prompts/wings4.txt -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/prune-ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/prune-ckpt.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/sample_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/sample_diffusion.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/README.md -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/eval_inpainting/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/eval_inpainting/launcher.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/eval_inpainting/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/eval_inpainting/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/resume_512/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/resume_512/launcher.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/resume_512/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/resume_512/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/resume_512_improvedaesthetic/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/resume_512_improvedaesthetic/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/resume_768_hr/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/resume_768_hr/launcher.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/resume_768_hr/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/resume_768_hr/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_edgeinpainting/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_edgeinpainting/launcher.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_edgeinpainting/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_edgeinpainting/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_iahr_torch111/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_iahr_torch111/launcher.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_iahr_torch111/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_iahr_torch111/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_iahr_torch111_ucg/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_iahr_torch111_ucg/launcher.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_iahr_torch111_ucg/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_iahr_torch111_ucg/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_improvedaesthetics/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_improvedaesthetics/launcher.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_improvedaesthetics/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_improvedaesthetics/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_laionhr_torch111/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_laionhr_torch111/launcher.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_laionhr_torch111/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v1_laionhr_torch111/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v2_laionhr1024/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v2_laionhr1024/launcher.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v2_laionhr1024/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v2_laionhr1024/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v2_laionhr1024_2/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v2_laionhr1024_2/launcher.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v2_laionhr1024_2/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v2_laionhr1024_2/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v2_pretraining/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v2_pretraining/launcher.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v2_pretraining/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v2_pretraining/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v3_pretraining/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v3_pretraining/launcher.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/slurm/v3_pretraining/sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/slurm/v3_pretraining/sbatch.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/test_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/test_gpu.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/test_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/test_gpu.sh -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/txt2img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/txt2img.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/scripts/vqgan_codebook_visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/scripts/vqgan_codebook_visualizer.py -------------------------------------------------------------------------------- /PyTorch/generative_models/stable-diffusion/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/generative_models/stable-diffusion/setup.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/LICENSE -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/README.md -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/create_pretraining_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/create_pretraining_data.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/BooksDownloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/BooksDownloader.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/BookscorpusTextFormatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/BookscorpusTextFormatting.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/Downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/Downloader.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/GooglePretrainedWeightDownloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/GooglePretrainedWeightDownloader.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/TextSharding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/TextSharding.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/WikiDownloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/WikiDownloader.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/WikicorpusTextFormatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/WikicorpusTextFormatting.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/__init__.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/bertPrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/bertPrep.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/create_datasets_from_start.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/data/create_datasets_from_start.sh -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/file_utils.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/lamb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/lamb.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/lamb_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/lamb_exp.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/lans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/lans.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/modeling.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/requirements.txt -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/run_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/run_pretraining.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/schedulers.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/bert_1.5b_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/bert_1.5b_config.json -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/bert_5b_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/bert_5b_config.json -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/deepspeed_config_bert_1.5b.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/deepspeed_config_bert_1.5b.json -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/deepspeed_config_bert_5b_lans.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/deepspeed_config_bert_5b_lans.json -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/hostsfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/hostsfile -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/run_bert_1.5b_8x.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/run_bert_1.5b_8x.sh -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/run_bert_5b_32x_lans.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/scripts/run_bert_5b_32x_lans.sh -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/tokenization.py -------------------------------------------------------------------------------- /PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/DeepSpeedExamples/deepspeed-bert/utils.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/LICENSE -------------------------------------------------------------------------------- /PyTorch/nlp/bert/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/README.md -------------------------------------------------------------------------------- /PyTorch/nlp/bert/bert_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/bert_config.json -------------------------------------------------------------------------------- /PyTorch/nlp/bert/bert_config_1.2B.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/bert_config_1.2B.json -------------------------------------------------------------------------------- /PyTorch/nlp/bert/create_pretraining_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/create_pretraining_data.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/data/BooksDownloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/data/BooksDownloader.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/data/BookscorpusTextFormatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/data/BookscorpusTextFormatting.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/data/Downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/data/Downloader.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/data/GooglePretrainedWeightDownloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/data/GooglePretrainedWeightDownloader.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/data/TextSharding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/data/TextSharding.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/data/WikiDownloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/data/WikiDownloader.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/data/WikicorpusTextFormatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/data/WikicorpusTextFormatting.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/data/__init__.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/data/bertPrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/data/bertPrep.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/data/create_datasets_from_start.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/data/create_datasets_from_start.sh -------------------------------------------------------------------------------- /PyTorch/nlp/bert/data/squad/squad_download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/data/squad/squad_download.sh -------------------------------------------------------------------------------- /PyTorch/nlp/bert/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/file_utils.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/lamb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/lamb.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/modeling.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/optimization.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/pack_pretraining_data_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/pack_pretraining_data_pytorch.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/pytorch_packed_data_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/pytorch_packed_data_checker.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/requirements.txt -------------------------------------------------------------------------------- /PyTorch/nlp/bert/run_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/run_pretraining.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/run_squad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/run_squad.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/schedulers.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/scripts/run_pretraining.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/scripts/run_pretraining.sh -------------------------------------------------------------------------------- /PyTorch/nlp/bert/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/tokenization.py -------------------------------------------------------------------------------- /PyTorch/nlp/bert/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/PyTorch/nlp/bert/utils.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/README.md -------------------------------------------------------------------------------- /ethics.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/ethics.md -------------------------------------------------------------------------------- /security.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HabanaAI/Model-References/HEAD/security.md --------------------------------------------------------------------------------