Metadata-Version: 2.1
Name: qai-hub-models
Version: 0.2.2
Summary: Models optimized for export to run on device.
Home-page: https://github.com/quic/ai-hub-models
Author: Qualcomm® Technologies, Inc.
License: MIT
Requires-Python: >=3.8, <3.11
License-File: LICENSE
Requires-Dist: Pillow ==10.0.1
Requires-Dist: gdown ==4.7.1
Requires-Dist: gitpython
Requires-Dist: huggingface-hub
Requires-Dist: numpy ==1.23.1
Requires-Dist: opencv-python ==4.8.1.78
Requires-Dist: prettytable
Requires-Dist: pytest ==7.4.2
Requires-Dist: pyyaml
Requires-Dist: qai-hub
Requires-Dist: requests
Requires-Dist: requests-toolbelt
Requires-Dist: schema
Requires-Dist: torch ==1.13.1
Requires-Dist: torchvision <=0.14.1
Requires-Dist: urllib3 <2
Provides-Extra: controlnet-quantized
Requires-Dist: transformers ==4.31.0 ; extra == 'controlnet-quantized'
Requires-Dist: diffusers[torch] ==0.21.4 ; extra == 'controlnet-quantized'
Requires-Dist: opencv-python ==4.8.1.78 ; extra == 'controlnet-quantized'
Provides-Extra: controlnet_quantized
Requires-Dist: transformers ==4.31.0 ; extra == 'controlnet_quantized'
Requires-Dist: diffusers[torch] ==0.21.4 ; extra == 'controlnet_quantized'
Requires-Dist: opencv-python ==4.8.1.78 ; extra == 'controlnet_quantized'
Provides-Extra: detr-resnet101
Requires-Dist: transformers ==4.31.0 ; extra == 'detr-resnet101'
Requires-Dist: timm ==0.9.7 ; extra == 'detr-resnet101'
Provides-Extra: detr-resnet101-dc5
Requires-Dist: transformers ==4.31.0 ; extra == 'detr-resnet101-dc5'
Requires-Dist: timm ==0.9.7 ; extra == 'detr-resnet101-dc5'
Provides-Extra: detr-resnet50
Requires-Dist: transformers ==4.31.0 ; extra == 'detr-resnet50'
Requires-Dist: timm ==0.9.7 ; extra == 'detr-resnet50'
Provides-Extra: detr-resnet50-dc5
Requires-Dist: transformers ==4.31.0 ; extra == 'detr-resnet50-dc5'
Requires-Dist: timm ==0.9.7 ; extra == 'detr-resnet50-dc5'
Provides-Extra: detr_resnet101
Requires-Dist: transformers ==4.31.0 ; extra == 'detr_resnet101'
Requires-Dist: timm ==0.9.7 ; extra == 'detr_resnet101'
Provides-Extra: detr_resnet101_dc5
Requires-Dist: transformers ==4.31.0 ; extra == 'detr_resnet101_dc5'
Requires-Dist: timm ==0.9.7 ; extra == 'detr_resnet101_dc5'
Provides-Extra: detr_resnet50
Requires-Dist: transformers ==4.31.0 ; extra == 'detr_resnet50'
Requires-Dist: timm ==0.9.7 ; extra == 'detr_resnet50'
Provides-Extra: detr_resnet50_dc5
Requires-Dist: transformers ==4.31.0 ; extra == 'detr_resnet50_dc5'
Requires-Dist: timm ==0.9.7 ; extra == 'detr_resnet50_dc5'
Provides-Extra: dev
Requires-Dist: coverage ==6.5.0 ; extra == 'dev'
Requires-Dist: pytest-xdist ==3.3.1 ; extra == 'dev'
Requires-Dist: pytest-cov ==4.1.0 ; extra == 'dev'
Requires-Dist: schema ==0.7.5 ; extra == 'dev'
Requires-Dist: pyyaml ==6.0.1 ; extra == 'dev'
Requires-Dist: types-PyYAML ; extra == 'dev'
Requires-Dist: types-requests ; extra == 'dev'
Requires-Dist: types-pillow ; extra == 'dev'
Requires-Dist: mypy ==0.991 ; extra == 'dev'
Requires-Dist: botocore ; extra == 'dev'
Requires-Dist: boto3 ; extra == 'dev'
Requires-Dist: ruamel-yaml ; extra == 'dev'
Requires-Dist: huggingface-hub ==0.20.3 ; extra == 'dev'
Requires-Dist: scikit-image >=0.21.0 ; extra == 'dev'
Requires-Dist: protobuf ==3.20.3 ; extra == 'dev'
Requires-Dist: tensorflow-cpu ==2.13.0 ; (sys_platform != "darwin") and extra == 'dev'
Requires-Dist: tensorflow-macos ==2.13.0 ; (sys_platform == "darwin") and extra == 'dev'
Provides-Extra: facebook-denoiser
Requires-Dist: denoiser ; extra == 'facebook-denoiser'
Requires-Dist: torchaudio ; extra == 'facebook-denoiser'
Requires-Dist: PySoundFile ; (sys_platform == "win32") and extra == 'facebook-denoiser'
Provides-Extra: facebook_denoiser
Requires-Dist: denoiser ; extra == 'facebook_denoiser'
Requires-Dist: torchaudio ; extra == 'facebook_denoiser'
Requires-Dist: PySoundFile ; (sys_platform == "win32") and extra == 'facebook_denoiser'
Provides-Extra: fastsam-s
Requires-Dist: ultralytics ==8.0.193 ; extra == 'fastsam-s'
Requires-Dist: torchvision ; extra == 'fastsam-s'
Provides-Extra: fastsam-x
Requires-Dist: ultralytics ==8.0.193 ; extra == 'fastsam-x'
Requires-Dist: torchvision ; extra == 'fastsam-x'
Provides-Extra: fastsam_s
Requires-Dist: ultralytics ==8.0.193 ; extra == 'fastsam_s'
Requires-Dist: torchvision ; extra == 'fastsam_s'
Provides-Extra: fastsam_x
Requires-Dist: ultralytics ==8.0.193 ; extra == 'fastsam_x'
Requires-Dist: torchvision ; extra == 'fastsam_x'
Provides-Extra: ffnet-122ns-lowres
Requires-Dist: scikit-image >=0.21.0 ; extra == 'ffnet-122ns-lowres'
Provides-Extra: ffnet-40s
Requires-Dist: scikit-image >=0.21.0 ; extra == 'ffnet-40s'
Provides-Extra: ffnet-54s
Requires-Dist: scikit-image >=0.21.0 ; extra == 'ffnet-54s'
Provides-Extra: ffnet-78s
Requires-Dist: scikit-image >=0.21.0 ; extra == 'ffnet-78s'
Provides-Extra: ffnet-78s-lowres
Requires-Dist: scikit-image >=0.21.0 ; extra == 'ffnet-78s-lowres'
Provides-Extra: ffnet_122ns_lowres
Requires-Dist: scikit-image >=0.21.0 ; extra == 'ffnet_122ns_lowres'
Provides-Extra: ffnet_40s
Requires-Dist: scikit-image >=0.21.0 ; extra == 'ffnet_40s'
Provides-Extra: ffnet_54s
Requires-Dist: scikit-image >=0.21.0 ; extra == 'ffnet_54s'
Provides-Extra: ffnet_78s
Requires-Dist: scikit-image >=0.21.0 ; extra == 'ffnet_78s'
Provides-Extra: ffnet_78s_lowres
Requires-Dist: scikit-image >=0.21.0 ; extra == 'ffnet_78s_lowres'
Provides-Extra: huggingface-wavlm-base-plus
Requires-Dist: transformers >=4.31.0 ; extra == 'huggingface-wavlm-base-plus'
Requires-Dist: soundfile >=0.12.1 ; extra == 'huggingface-wavlm-base-plus'
Requires-Dist: librosa >=0.10.1 ; extra == 'huggingface-wavlm-base-plus'
Requires-Dist: datasets >=2.14.5 ; extra == 'huggingface-wavlm-base-plus'
Provides-Extra: huggingface_wavlm_base_plus
Requires-Dist: transformers >=4.31.0 ; extra == 'huggingface_wavlm_base_plus'
Requires-Dist: soundfile >=0.12.1 ; extra == 'huggingface_wavlm_base_plus'
Requires-Dist: librosa >=0.10.1 ; extra == 'huggingface_wavlm_base_plus'
Requires-Dist: datasets >=2.14.5 ; extra == 'huggingface_wavlm_base_plus'
Provides-Extra: lama-dilated
Requires-Dist: matplotlib ; extra == 'lama-dilated'
Requires-Dist: pandas ; extra == 'lama-dilated'
Requires-Dist: albumentations ==0.5.2 ; extra == 'lama-dilated'
Requires-Dist: pytorch-lightning ==1.6.0 ; extra == 'lama-dilated'
Requires-Dist: webdataset ; extra == 'lama-dilated'
Requires-Dist: easydict ==1.10 ; extra == 'lama-dilated'
Requires-Dist: kornia ==0.5.0 ; extra == 'lama-dilated'
Requires-Dist: hydra-core ==1.3.0 ; extra == 'lama-dilated'
Requires-Dist: omegaconf ==2.3.0 ; extra == 'lama-dilated'
Requires-Dist: scikit-learn ==1.3.0 ; extra == 'lama-dilated'
Provides-Extra: lama_dilated
Requires-Dist: matplotlib ; extra == 'lama_dilated'
Requires-Dist: pandas ; extra == 'lama_dilated'
Requires-Dist: albumentations ==0.5.2 ; extra == 'lama_dilated'
Requires-Dist: pytorch-lightning ==1.6.0 ; extra == 'lama_dilated'
Requires-Dist: webdataset ; extra == 'lama_dilated'
Requires-Dist: easydict ==1.10 ; extra == 'lama_dilated'
Requires-Dist: kornia ==0.5.0 ; extra == 'lama_dilated'
Requires-Dist: hydra-core ==1.3.0 ; extra == 'lama_dilated'
Requires-Dist: omegaconf ==2.3.0 ; extra == 'lama_dilated'
Requires-Dist: scikit-learn ==1.3.0 ; extra == 'lama_dilated'
Provides-Extra: litehrnet
Requires-Dist: mmpose <=1.2.0 ; extra == 'litehrnet'
Requires-Dist: mmcv ==2.1.0 ; extra == 'litehrnet'
Requires-Dist: mmdet <=3.2.0 ; extra == 'litehrnet'
Provides-Extra: mediapipe-face
Requires-Dist: opencv-python ; extra == 'mediapipe-face'
Requires-Dist: requests ; extra == 'mediapipe-face'
Provides-Extra: mediapipe-hand
Requires-Dist: opencv-python ; extra == 'mediapipe-hand'
Requires-Dist: requests ; extra == 'mediapipe-hand'
Provides-Extra: mediapipe-pose
Requires-Dist: opencv-python ; extra == 'mediapipe-pose'
Requires-Dist: requests ; extra == 'mediapipe-pose'
Provides-Extra: mediapipe-selfie
Requires-Dist: tflite ==2.10.0 ; extra == 'mediapipe-selfie'
Provides-Extra: mediapipe_face
Requires-Dist: opencv-python ; extra == 'mediapipe_face'
Requires-Dist: requests ; extra == 'mediapipe_face'
Provides-Extra: mediapipe_hand
Requires-Dist: opencv-python ; extra == 'mediapipe_hand'
Requires-Dist: requests ; extra == 'mediapipe_hand'
Provides-Extra: mediapipe_pose
Requires-Dist: opencv-python ; extra == 'mediapipe_pose'
Requires-Dist: requests ; extra == 'mediapipe_pose'
Provides-Extra: mediapipe_selfie
Requires-Dist: tflite ==2.10.0 ; extra == 'mediapipe_selfie'
Provides-Extra: mobiledet
Requires-Dist: tucker-conv ==1.0.1 ; extra == 'mobiledet'
Requires-Dist: timm ==0.6.13 ; extra == 'mobiledet'
Provides-Extra: openai-clip
Requires-Dist: torchvision ; extra == 'openai-clip'
Requires-Dist: ftfy ==6.1.1 ; extra == 'openai-clip'
Requires-Dist: regex ==2023.10.3 ; extra == 'openai-clip'
Provides-Extra: openai_clip
Requires-Dist: torchvision ; extra == 'openai_clip'
Requires-Dist: ftfy ==6.1.1 ; extra == 'openai_clip'
Requires-Dist: regex ==2023.10.3 ; extra == 'openai_clip'
Provides-Extra: openpose
Requires-Dist: scipy ; extra == 'openpose'
Requires-Dist: matplotlib ; extra == 'openpose'
Provides-Extra: real-esrgan-general-x4v3
Requires-Dist: opencv-python ; extra == 'real-esrgan-general-x4v3'
Requires-Dist: PyYAML ; extra == 'real-esrgan-general-x4v3'
Requires-Dist: requests ; extra == 'real-esrgan-general-x4v3'
Requires-Dist: scipy ; extra == 'real-esrgan-general-x4v3'
Requires-Dist: seaborn ; extra == 'real-esrgan-general-x4v3'
Requires-Dist: basicsr ; extra == 'real-esrgan-general-x4v3'
Provides-Extra: real-esrgan-x4plus
Requires-Dist: opencv-python ; extra == 'real-esrgan-x4plus'
Requires-Dist: scipy ; extra == 'real-esrgan-x4plus'
Requires-Dist: seaborn ; extra == 'real-esrgan-x4plus'
Requires-Dist: basicsr ; extra == 'real-esrgan-x4plus'
Provides-Extra: real_esrgan_general_x4v3
Requires-Dist: opencv-python ; extra == 'real_esrgan_general_x4v3'
Requires-Dist: PyYAML ; extra == 'real_esrgan_general_x4v3'
Requires-Dist: requests ; extra == 'real_esrgan_general_x4v3'
Requires-Dist: scipy ; extra == 'real_esrgan_general_x4v3'
Requires-Dist: seaborn ; extra == 'real_esrgan_general_x4v3'
Requires-Dist: basicsr ; extra == 'real_esrgan_general_x4v3'
Provides-Extra: real_esrgan_x4plus
Requires-Dist: opencv-python ; extra == 'real_esrgan_x4plus'
Requires-Dist: scipy ; extra == 'real_esrgan_x4plus'
Requires-Dist: seaborn ; extra == 'real_esrgan_x4plus'
Requires-Dist: basicsr ; extra == 'real_esrgan_x4plus'
Provides-Extra: sam
Requires-Dist: matplotlib ; extra == 'sam'
Requires-Dist: opencv-python ; extra == 'sam'
Requires-Dist: pycocotools ; extra == 'sam'
Requires-Dist: requests ; extra == 'sam'
Provides-Extra: stable-diffusion
Requires-Dist: transformers ==4.31.0 ; extra == 'stable-diffusion'
Requires-Dist: coremltools ==6.2 ; extra == 'stable-diffusion'
Requires-Dist: diffusers[torch] ==0.21.4 ; extra == 'stable-diffusion'
Provides-Extra: stable-diffusion-quantized
Requires-Dist: transformers ==4.31.0 ; extra == 'stable-diffusion-quantized'
Requires-Dist: diffusers[torch] ==0.21.4 ; extra == 'stable-diffusion-quantized'
Provides-Extra: stable_diffusion
Requires-Dist: transformers ==4.31.0 ; extra == 'stable_diffusion'
Requires-Dist: coremltools ==6.2 ; extra == 'stable_diffusion'
Requires-Dist: diffusers[torch] ==0.21.4 ; extra == 'stable_diffusion'
Provides-Extra: stable_diffusion_quantized
Requires-Dist: transformers ==4.31.0 ; extra == 'stable_diffusion_quantized'
Requires-Dist: diffusers[torch] ==0.21.4 ; extra == 'stable_diffusion_quantized'
Provides-Extra: stylegan2
Requires-Dist: click >=8.0 ; extra == 'stylegan2'
Provides-Extra: trocr
Requires-Dist: transformers ==4.33.2 ; extra == 'trocr'
Requires-Dist: sentencepiece ; extra == 'trocr'
Provides-Extra: whisper-asr
Requires-Dist: openai-whisper ==20230314 ; extra == 'whisper-asr'
Requires-Dist: scipy ; extra == 'whisper-asr'
Provides-Extra: whisper_asr
Requires-Dist: openai-whisper ==20230314 ; extra == 'whisper_asr'
Requires-Dist: scipy ; extra == 'whisper_asr'
Provides-Extra: yolov7
Requires-Dist: matplotlib ; extra == 'yolov7'
Requires-Dist: opencv-python ; extra == 'yolov7'
Requires-Dist: PyYAML ; extra == 'yolov7'
Requires-Dist: requests ; extra == 'yolov7'
Requires-Dist: scipy ; extra == 'yolov7'
Requires-Dist: seaborn ; extra == 'yolov7'
Provides-Extra: yolov8-det
Requires-Dist: ultralytics ==8.0.193 ; extra == 'yolov8-det'
Provides-Extra: yolov8-seg
Requires-Dist: ultralytics ==8.0.193 ; extra == 'yolov8-seg'
Provides-Extra: yolov8_det
Requires-Dist: ultralytics ==8.0.193 ; extra == 'yolov8_det'
Provides-Extra: yolov8_seg
Requires-Dist: ultralytics ==8.0.193 ; extra == 'yolov8_seg'

Models optimized for export to run on device.
