clarifai 9.10.2__py3-none-any.whl → 9.10.3__py3-none-any.whl

clarifai/models/model_serving/examples/text_classification/xlm-roberta/requirements.txt

@@ -1,7 +0,0 @@
-clarifai>9.5.3 # for model upload features
-tritonclient[all]
-torch==1.13.1
-transformers==4.30.2
-scipy==1.10.1
-sentencepiece==0.1.99
-protobuf<4.21.3

clarifai/models/model_serving/examples/text_embedding/README.md

@@ -1,9 +0,0 @@
-## Text Embedding Triton Model Examples
-
-These can be used on the fly with minimal or no changes to test deploy text embedding models to the Clarifai platform. See the required files section for each model below.
-
-* ### [Instructor-xl](https://huggingface.co/hkunlp/instructor-xl)
-
-	Requirements to run tests locally:
-
-	* Download/Clone the [huggingface model](https://huggingface.co/hkunlp/instructor-xl) into the **instructor-xl/1/** directory then start the triton server.

clarifai/models/model_serving/examples/text_to_image/README.md

@@ -1,9 +0,0 @@
-## Text to Image Triton Model Examples
-
-These can be used on the fly with minimal or no changes to test deploy text to image models to the Clarifai platform. See the required files section for each model below.
-
-* ### [sd-v1.5 (Stable-Diffusion-v1.5)](./sd-v1.5/)
-
-	Requirements to run tests locally:
-
-	* Download/Clone the [huggingface model](https://huggingface.co/runwayml/stable-diffusion-v1-5) into the **sd-v1.5/1/** directory then start the triton server.

clarifai/models/model_serving/examples/text_to_image/sd-v1.5/1/inference.py

@@ -1,52 +0,0 @@
-# This file contains boilerplate code to allow users write their model
-# inference code that will then interact with the Triton Inference Server
-# Python backend to serve end user requests.
-# The module name, module path, class name & get_predictions() method names MUST be maintained as is
-# but other methods may be added within the class as deemed fit provided
-# they are invoked within the main get_predictions() inference method
-# if they play a role in any step of model inference
-"""User model inference script."""
-
-import os
-from pathlib import Path
-
-import numpy as np
-import torch
-from diffusers import StableDiffusionPipeline
-
-from clarifai.models.model_serving.models.model_types import text_to_image
-from clarifai.models.model_serving.models.output import ImageOutput
-
-
-class InferenceModel:
-  """User model inference class."""
-
-  def __init__(self) -> None:
-    """
-    Load inference time artifacts that are called frequently .e.g. models, tokenizers, etc.
-    in this method so they are loaded only once for faster inference.
-    """
-    self.base_path: Path = os.path.dirname(__file__)
-    self.huggingface_model_path = os.path.join(self.base_path, "stable-diffusion-v1-5")
-    self.device = "cuda" if torch.cuda.is_available() else "cpu"
-    self.pipeline = StableDiffusionPipeline.from_pretrained(
-        self.huggingface_model_path, torch_dtype=torch.float16)
-    self.pipeline = self.pipeline.to(self.device)
-
-  @text_to_image
-  def get_predictions(self, input_data):
-    """
-    Main model inference method.
-
-    Args:
-    -----
-      input_data: A single input data item to predict on.
-        Input data can be an image or text, etc depending on the model type.
-
-    Returns:
-    --------
-      One of the clarifai.models.model_serving.models.output types. Refer to the README/docs
-    """
-    out_image = self.pipeline(input_data).images[0]
-    out_image = np.asarray(out_image)
-    return ImageOutput(image=out_image)

clarifai/models/model_serving/examples/text_to_image/sd-v1.5/1/model.py

@@ -1,60 +0,0 @@
-# Copyright 2023 Clarifai, Inc.
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-"""Triton inference server Python Backend Model."""
-
-import os
-import sys
-
-try:
-  import triton_python_backend_utils as pb_utils
-except ModuleNotFoundError:
-  pass
-from google.protobuf import text_format
-from tritonclient.grpc.model_config_pb2 import ModelConfig
-
-
-class TritonPythonModel:
-  """
-  Triton Python BE Model.
-  """
-
-  def initialize(self, args):
-    """
-    Triton server init.
-    """
-    args["model_repository"] = args["model_repository"].replace("/1/model.py", "")
-    sys.path.append(os.path.dirname(__file__))
-    from inference import InferenceModel
-
-    self.inference_obj = InferenceModel()
-
-    # Read input_name from config file
-    self.config_msg = ModelConfig()
-    with open(os.path.join(args["model_repository"], "config.pbtxt"), "r") as f:
-      cfg = f.read()
-    text_format.Merge(cfg, self.config_msg)
-    self.input_name = [inp.name for inp in self.config_msg.input][0]
-
-  def execute(self, requests):
-    """
-    Serve model inference requests.
-    """
-    responses = []
-
-    for request in requests:
-      in_batch = pb_utils.get_input_tensor_by_name(request, self.input_name)
-      in_batch = in_batch.as_numpy()
-      inference_response = self.inference_obj.get_predictions(in_batch)
-      responses.append(inference_response)
-
-    return responses

clarifai/models/model_serving/examples/text_to_image/sd-v1.5/config.pbtxt

@@ -1,22 +0,0 @@
-name: "sd-v1.5"
-max_batch_size: 1
-input {
-  name: "text"
-  data_type: TYPE_STRING
-  dims: 1
-}
-output {
-  name: "image"
-  data_type: TYPE_UINT8
-  dims: -1
-  dims: -1
-  dims: 3
-}
-instance_group {
-  count: 1
-  kind: KIND_GPU
-}
-dynamic_batching {
-  max_queue_delay_microseconds: 500
-}
-backend: "python"

clarifai/models/model_serving/examples/text_to_image/sd-v1.5/requirements.txt

@@ -1,6 +0,0 @@
-clarifai>9.5.3
-tritonclient[all]
-torch==1.13.1
-transformers==4.30.2
-Pillow==10.0.0
-diffusers==0.19.0

clarifai/models/model_serving/examples/text_to_text/README.md

@@ -1,10 +0,0 @@
-## Text-to-Text Triton Model Examples
-
-These can be used on the fly with minimal or no changes to test deploy all models that take a text input and yield a text output prediction e.g. text generation, summarization and translation models to the Clarifai platform. See the required files section for each model below.
-
-* ### [Bart-paper2slides-summarizer](https://huggingface.co/com3dian/Bart-large-paper2slides-summarizer)
-
-	Requirements to run tests locally:
-
-	* Download/Clone the [huggingface model](https://huggingface.co/com3dian/Bart-large-paper2slides-summarizer) and store it under the **bart-summarize/1/** directory.
-	* Rename the downloaded folder to **bart-large-summarizer** OR change the **self.huggingface_model_path** attribute in the [inference.py script](./bart-summarize/1/inference.py) to match the folder name

clarifai/models/model_serving/examples/text_to_text/bart-summarize/config.pbtxt

@@ -1,20 +0,0 @@
-name: "bart-summarize"
-max_batch_size: 1
-input {
-  name: "text"
-  data_type: TYPE_STRING
-  dims: 1
-}
-output {
-  name: "text"
-  data_type: TYPE_STRING
-  dims: 1
-}
-instance_group {
-  count: 1
-  kind: KIND_GPU
-}
-dynamic_batching {
-  max_queue_delay_microseconds: 500
-}
-backend: "python"

clarifai/models/model_serving/examples/text_to_text/bart-summarize/requirements.txt

@@ -1,4 +0,0 @@
-clarifai>9.5.3
-tritonclient[all]
-torch==1.13.1
-transformers==4.30.2

clarifai/models/model_serving/examples/visual_detection/README.md

@@ -1,11 +0,0 @@
-## Visual Detection Triton Model Examples
-
-These can be used on the fly with minimal or no changes to test deploy visual detection models to the Clarifai platform. See the required files section for each model below.
-
-* ### [Yolov5x](./yolov5x/)
-
-	Required files (not included here due to upload size limits):
-
-	* Download the yolov5x folder from above.
-	* Download the `Yolov5 repo` and the `yolov5-x checkpoint` and store them under the `1/` directory of the yolov5x folder.
-	* zip and test deploy to your Clarifai app

clarifai/models/model_serving/examples/visual_detection/yolov5x/config.pbtxt

@@ -1,36 +0,0 @@
-name: "yolov5_test"
-max_batch_size: 1
-input {
-  name: "image"
-  data_type: TYPE_UINT8
-  dims: -1
-  dims: -1
-  dims: 3
-}
-output {
-  name: "predicted_bboxes"
-  data_type: TYPE_FP32
-  dims: -1
-  dims: 4
-}
-output {
-  name: "predicted_labels"
-  data_type: TYPE_INT32
-  dims: -1
-  dims: 1
-  label_filename: "labels.txt"
-}
-output {
-  name: "predicted_scores"
-  data_type: TYPE_FP32
-  dims: -1
-  dims: 1
-}
-instance_group {
-  count: 1
-  kind: KIND_GPU
-}
-dynamic_batching {
-  max_queue_delay_microseconds: 500
-}
-backend: "python"

clarifai/models/model_serving/examples/visual_detection/yolov5x/labels.txt

@@ -1,80 +0,0 @@
-person
-bicycle
-car
-motorcycle
-airplane
-bus
-train
-truck
-boat
-traffic-light
-fire-hydrant
-stop-sign
-parking-meter
-bench
-bird
-cat
-dog
-horse
-sheep
-cow
-elephant
-bear
-zebra
-giraffe
-backpack
-umbrella
-handbag
-tie
-suitcase
-frisbee
-skis
-snowboard
-sports-ball
-kite
-baseball-bat
-baseball-glove
-skateboard
-surfboard
-tennis-racket
-bottle
-wine-glass
-cup
-fork
-knife
-spoon
-bowl
-banana
-apple
-sandwich
-orange
-broccoli
-carrot
-hot-dog
-pizza
-donut
-cake
-chair
-couch
-potted-plant
-bed
-dining-table
-toilet
-tv
-laptop
-mouse
-remote
-keyboard
-cell-phone
-microwave
-oven
-toaster
-sink
-refrigerator
-book
-clock
-vase
-scissors
-teddy-bear
-hair-drier
-toothbrush

clarifai/models/model_serving/examples/visual_detection/yolov5x/requirements.txt

@@ -1,12 +0,0 @@
-# YOLOv5 requirements
-tritonclient[all]
-clarifai>9.5.3 # for model upload features
-matplotlib>=3.2.2
-opencv-python>=4.1.1
-Pillow>=7.1.2
-PyYAML>=5.3.1
-torch>=1.7.0,<2.0
-torchvision>=0.8.1
-protobuf<4.21.3  # https://github.com/ultralytics/yolov5/issues/8012
-pandas>=1.1.4
-seaborn>=0.11.0

clarifai/models/model_serving/examples/visual_embedding/README.md

@@ -1,9 +0,0 @@
-## Visual Embedding Triton Model Examples
-
-These can be used on the fly with minimal or no changes to test deploy visual embedding models to the Clarifai platform. See the required files section for each model below.
-
-* ### [vit-base](./vit-base/)
-
-	Requirements to run tests locally:
-
-	* Download/Clone the [huggingface model](https://huggingface.co/google/vit-base-patch16-224) into the **vit-base/1/** directory then start the triton server.

clarifai/models/model_serving/examples/visual_embedding/vit-base/config.pbtxt

@@ -1,22 +0,0 @@
-name: "vit-base"
-max_batch_size: 1
-input {
-  name: "image"
-  data_type: TYPE_UINT8
-  dims: -1
-  dims: -1
-  dims: 3
-}
-output {
-  name: "embeddings"
-  data_type: TYPE_FP32
-  dims: -1
-}
-instance_group {
-  count: 1
-  kind: KIND_GPU
-}
-dynamic_batching {
-  max_queue_delay_microseconds: 500
-}
-backend: "python"

clarifai/models/model_serving/examples/visual_embedding/vit-base/requirements.txt

@@ -1,5 +0,0 @@
-clarifai>9.5.3
-tritonclient[all]
-torch==1.13.1
-transformers==4.30.2
-Pillow==10.0.0

clarifai/models/model_serving/examples/visual_segmentation/README.md

@@ -1,9 +0,0 @@
-## Visual Segmentation Triton Model Examples
-
-These can be used on the fly with minimal or no changes to test deploy visual segmentation models to the Clarifai platform. See the required files section for each model below.
-
-* ### [segformer-b2](./segformer-b2/)
-
-	Requirements to run tests locally:
-
-	* Download/Clone the [huggingface model](https://huggingface.co/mattmdjaga/segformer_b2_clothes) into the **segformer-b2/1/** directory then start the triton server.

clarifai/models/model_serving/examples/visual_segmentation/segformer-b2/config.pbtxt

@@ -1,24 +0,0 @@
-name: "segformer-b2"
-max_batch_size: 1
-input {
-  name: "image"
-  data_type: TYPE_UINT8
-  dims: -1
-  dims: -1
-  dims: 3
-}
-output {
-  name: "predicted_mask"
-  data_type: TYPE_INT64
-  dims: -1
-  dims: -1
-  label_filename: "labels.txt"
-}
-instance_group {
-  count: 1
-  kind: KIND_GPU
-}
-dynamic_batching {
-  max_queue_delay_microseconds: 500
-}
-backend: "python"

clarifai/models/model_serving/examples/visual_segmentation/segformer-b2/labels.txt

@@ -1,18 +0,0 @@
-background
-hat
-hair
-sunglass
-upper-clothes
-skirt
-pants
-dress
-belt
-left-shoe
-right-shoe
-face
-left-leg
-right-leg
-left-arm
-right-arm
-bag
-scarf

clarifai/models/model_serving/examples/visual_segmentation/segformer-b2/requirements.txt

@@ -1,5 +0,0 @@
-torch==1.13.1
-clarifai>9.5.3
-tritonclient[all]
-transformers==4.30.2
-Pillow==10.0.0

clarifai/models/model_serving/model_config/model_types_config/multimodal-embedder.yaml

@@ -1,24 +0,0 @@
-triton:
-  input:
-    - name: image
-      data_type: TYPE_UINT8
-      dims: [-1, -1, 3]
-      optional: true
-    - name: text
-      data_type: TYPE_STRING
-      dims: [1]
-      optional: true
-  output:
-    - name: embeddings
-      data_type: TYPE_FP32
-      dims: [-1]
-      labels: false
-inference:
-  wrap_func: multimodal_embedder
-  return_type: EmbeddingOutput
-field_maps:
-  input_fields_map:
-    image: image
-    text: text
-  output_fields_map:
-    embeddings: embeddings

clarifai/models/model_serving/model_config/model_types_config/text-classifier.yaml

@@ -1,18 +0,0 @@
-triton:
-  input:
-    - name: text
-      data_type: TYPE_STRING
-      dims: [1]
-  output:
-    - name: softmax_predictions
-      data_type: TYPE_FP32
-      dims: [-1]
-      labels: true
-inference:
-  wrap_func: text_classifier
-  return_type: ClassifierOutput
-field_maps:
-  input_fields_map:
-    text: text
-  output_fields_map:
-    concepts: softmax_predictions

clarifai/models/model_serving/model_config/model_types_config/text-embedder.yaml

@@ -1,18 +0,0 @@
-triton:
-  input:
-    - name: text
-      data_type: TYPE_STRING
-      dims: [1]
-  output:
-    - name: embeddings
-      data_type: TYPE_FP32
-      dims: [-1]
-      labels: false
-inference:
-  wrap_func: text_embedder
-  return_type: EmbeddingOutput
-field_maps:
-  input_fields_map:
-    text: text
-  output_fields_map:
-    embeddings: embeddings

clarifai/models/model_serving/model_config/model_types_config/text-to-image.yaml

@@ -1,18 +0,0 @@
-triton:
-  input:
-    - name: text
-      data_type: TYPE_STRING
-      dims: [1]
-  output:
-    - name: image
-      data_type: TYPE_UINT8
-      dims: [-1, -1, 3]
-      labels: false
-inference:
-  wrap_func: text_to_image
-  return_type: ImageOutput
-field_maps:
-  input_fields_map:
-    text: text
-  output_fields_map:
-    image: image

clarifai/models/model_serving/model_config/model_types_config/text-to-text.yaml

@@ -1,18 +0,0 @@
-triton:
-  input:
-    - name: text
-      data_type: TYPE_STRING
-      dims: [1]
-  output:
-    - name: text
-      data_type: TYPE_STRING
-      dims: [1]
-      labels: false
-inference:
-  wrap_func: text_to_text
-  return_type: TextOutput
-field_maps:
-  input_fields_map:
-    text: text
-  output_fields_map:
-    text: text

clarifai/models/model_serving/model_config/model_types_config/visual-classifier.yaml

@@ -1,18 +0,0 @@
-triton:
-  input:
-    - name: image
-      data_type: TYPE_UINT8
-      dims: [-1, -1, 3]
-  output:
-    - name: softmax_predictions
-      data_type: TYPE_FP32
-      dims: [-1]
-      labels: true
-inference:
-  wrap_func: visual_classifier
-  return_type: ClassifierOutput
-field_maps:
-  input_fields_map:
-    image: image
-  output_fields_map:
-    concepts: softmax_predictions

clarifai/models/model_serving/model_config/model_types_config/visual-detector.yaml

@@ -1,28 +0,0 @@
-triton:
-  input:
-    - name: image
-      data_type: TYPE_UINT8
-      dims: [-1, -1, 3]
-  output:
-    - name: predicted_bboxes
-      data_type: TYPE_FP32
-      dims: [-1, 4]
-      labels: false
-    - name: predicted_labels
-      data_type: TYPE_INT32
-      dims: [-1, 1]
-      labels: true
-    - name: predicted_scores
-      data_type: TYPE_FP32
-      dims: [-1, 1]
-      labels: false
-inference:
-  wrap_func: visual_detector
-  return_type: VisualDetectorOutput
-field_maps:
-  input_fields_map:
-    image: image
-  output_fields_map:
-    "regions[...].region_info.bounding_box": "predicted_bboxes"
-    "regions[...].data.concepts[...].id": "predicted_labels"
-    "regions[...].data.concepts[...].value": "predicted_scores"

clarifai/models/model_serving/model_config/model_types_config/visual-embedder.yaml

@@ -1,18 +0,0 @@
-triton:
-  input:
-    - name: image
-      data_type: TYPE_UINT8
-      dims: [-1, -1, 3]
-  output:
-    - name: embeddings
-      data_type: TYPE_FP32
-      dims: [-1]
-      labels: false
-inference:
-  wrap_func: visual_embedder
-  return_type: EmbeddingOutput
-field_maps:
-  input_fields_map:
-    image: image
-  output_fields_map:
-    embeddings: embeddings

clarifai/models/model_serving/model_config/model_types_config/visual-segmenter.yaml

@@ -1,18 +0,0 @@
-triton:
-  input:
-    - name: image
-      data_type: TYPE_UINT8
-      dims: [-1, -1, 3]
-  output:
-    - name: predicted_mask
-      data_type: TYPE_INT64
-      dims: [-1, -1]
-      labels: true
-inference:
-  wrap_func: visual_segmenter
-  return_type: MasksOutput
-field_maps:
-  input_fields_map:
-    image: image
-  output_fields_map:
-    "regions[...].region_info.mask,regions[...].data.concepts": "predicted_mask"

clarifai/modules/README.md

@@ -1,5 +0,0 @@
-# Module Utils
-
-Additional helper functions for creating Clarifai Modules should be placed here so that they can be reused across modules.
-
-This should still not import streamlit as we want to keep clarifai-python-utils lightweight. If you find we need utilities for streamlit itself we should start a new repo for that. Please contact support@clarifai.com to do so.