clarifai 9.11.1__py3-none-any.whl → 10.0.0__py3-none-any.whl

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

@@ -1,12 +0,0 @@
-## Image Classification Triton Model Examples
-
-These can be used on the fly with minimal or no changes to test deploy image classification models to the Clarifai platform. See the required files section for each model below.
-
-* ### [VIT Age Classifier](./age_vit/)
-
-	Required files to run tests locally:
-
-	* Download the [model checkpoint from huggingface](https://huggingface.co/nateraw/vit-age-classifier/tree/main) and store it under `age_vit/1/checkpoint/`
-	```
-	huggingface-cli download nateraw/vit-age-classifier --local-dir age_vit/1/checkpoint/ --local-dir-use-symlinks False
-	```

clarifai/models/model_serving/examples/image_classification/age_vit/1/inference.py

@@ -1,64 +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
-from typing import Callable
-
-import torch
-from scipy.special import softmax
-from transformers import AutoImageProcessor, ViTForImageClassification
-
-from clarifai.models.model_serving.model_config import ModelTypes, get_model_config
-from clarifai.models.model_serving.models.output import ClassifierOutput
-
-config = get_model_config(ModelTypes.visual_classifier)
-
-
-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: Path = os.path.join(self.base_path, "checkpoint")
-    self.transforms = AutoImageProcessor.from_pretrained(self.huggingface_model_path)
-    self.model: Callable = ViTForImageClassification.from_pretrained(self.huggingface_model_path)
-    self.device = "cuda:0" if torch.cuda.is_available() else "cpu"
-
-  @config.inference.wrap_func
-  def get_predictions(self, input_data: list, **kwargs) -> list:
-    """
-    Main model inference method.
-
-    Args:
-    -----
-      input_data: A list of input data item to predict on.
-        Input data can be an image or text, etc depending on the model type.
-
-      **kwargs: your inference parameters.
-
-    Returns:
-    --------
-      List of one of the `clarifai.models.model_serving.models.output types` or `config.inference.return_type(your_output)`. Refer to the README/docs
-    """
-    # Transform image and pass it to the model
-    inputs = self.transforms(input_data, return_tensors='pt')
-    with torch.no_grad():
-      preds = self.model(**inputs).logits
-    outputs = []
-    for pred in preds:
-      pred_scores = softmax(
-          pred.detach().numpy())  # alt: softmax(output.logits[0].detach().numpy())
-      outputs.append(ClassifierOutput(predicted_scores=pred_scores))
-
-    return outputs

clarifai/models/model_serving/examples/image_classification/age_vit/1/model.py

@@ -1,74 +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
-from clarifai.models.model_serving.model_config.inference_parameter import parse_req_parameters
-
-
-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_names = [inp.name for inp in self.config_msg.input]
-
-  def execute(self, requests):
-    """
-    Serve model inference requests.
-    """
-    responses = []
-
-    for request in requests:
-      parameters = request.parameters()
-      parameters = parse_req_parameters(parameters) if parameters else {}
-
-      if len(self.input_names) == 1:
-        in_batch = pb_utils.get_input_tensor_by_name(request, self.input_names[0])
-        in_batch = in_batch.as_numpy()
-        inference_response = self.inference_obj.get_predictions(in_batch, **parameters)
-      else:
-        multi_in_batch_dict = {}
-        for input_name in self.input_names:
-          in_batch = pb_utils.get_input_tensor_by_name(request, input_name)
-          in_batch = in_batch.as_numpy() if in_batch is not None else []
-          multi_in_batch_dict.update({input_name: in_batch})
-
-        inference_response = self.inference_obj.get_predictions(multi_in_batch_dict, **parameters)
-
-      responses.append(inference_response)
-
-    return responses

clarifai/models/model_serving/examples/image_classification/age_vit/1/vit-age-classifier/README.md

@@ -1,11 +0,0 @@
----
-tags:
-- image-classification
-- pytorch
-datasets:
-- fairface
----
-
-## ViT For Age Classification
-
-A vision transformer finetuned to classify the age of a given person's face.

clarifai/models/model_serving/examples/image_classification/age_vit/1/vit-age-classifier/config.json

@@ -1,42 +0,0 @@
-{
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
-  "architectures": [
-    "ViTForImageClassification"
-  ],
-  "attention_probs_dropout_prob": 0.0,
-  "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.0,
-  "hidden_size": 768,
-  "id2label": {
-    "0": "0-2",
-    "1": "3-9",
-    "2": "10-19",
-    "3": "20-29",
-    "4": "30-39",
-    "5": "40-49",
-    "6": "50-59",
-    "7": "60-69",
-    "8": "more than 70"
-  },
-  "image_size": 224,
-  "initializer_range": 0.02,
-  "intermediate_size": 3072,
-  "label2id": {
-    "0-2": 0,
-    "3-9": 1,
-    "10-19": 2,
-    "20-29": 3,
-    "30-39": 4,
-    "40-49": 5,
-    "50-59": 6,
-    "60-69": 7,
-    "more than 70": 8
-  },
-  "layer_norm_eps": 1e-12,
-  "model_type": "vit",
-  "num_attention_heads": 12,
-  "num_channels": 3,
-  "num_hidden_layers": 12,
-  "patch_size": 16,
-  "transformers_version": "4.5.0.dev0"
-}

clarifai/models/model_serving/examples/image_classification/age_vit/1/vit-age-classifier/preprocessor_config.json

@@ -1,15 +0,0 @@
-{
-    "do_normalize": true,
-    "do_resize": true,
-    "image_mean": [
-        0.5,
-        0.5,
-        0.5
-    ],
-    "image_std": [
-        0.5,
-        0.5,
-        0.5
-    ],
-    "size": 224
-}

clarifai/models/model_serving/examples/image_classification/age_vit/config.pbtxt

@@ -1,23 +0,0 @@
-name: "age_vit"
-max_batch_size: 1
-input {
-  name: "image"
-  data_type: TYPE_UINT8
-  dims: -1
-  dims: -1
-  dims: 3
-}
-output {
-  name: "softmax_predictions"
-  data_type: TYPE_FP32
-  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/image_classification/age_vit/labels.txt

@@ -1,9 +0,0 @@
-0-2
-3-9
-10-19
-20-29
-30-39
-40-49
-50-59
-60-69
-more than 70

clarifai/models/model_serving/examples/image_classification/age_vit/requirements.txt

@@ -1,7 +0,0 @@
-clarifai>9.10.5
-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/multimodal_embedder/README.md

@@ -1,12 +0,0 @@
-## Multimodal Embedder Triton Model Examples
-
-These can be used on the fly with minimal or no changes to test deploy image classification models to the Clarifai platform. See the required files section for each model below.
-
-* ### [CLIP](./clip/)
-
-	Required files to run tests locally:
-
-	* Download the [model checkpoint from huggingface](https://huggingface.co/openai/clip-vit-base-patch32) and store it under `clip/1/checkpoint/`
-	```
-	huggingface-cli download openai/clip-vit-base-patch32 --local-dir clip/1/checkpoint/ --local-dir-use-symlinks False --exclude *.msgpack *.h5
-	```

clarifai/models/model_serving/examples/multimodal_embedder/clip/1/inference.py

@@ -1,66 +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 torch
-from transformers import CLIPModel, CLIPProcessor
-from clarifai.models.model_serving.model_config import ModelTypes, get_model_config
-
-config = get_model_config(ModelTypes.multimodal_embedder)
-
-
-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__)
-    ## sample model loading code:
-    #self.checkpoint_path: Path = os.path.join(self.base_path, "your checkpoint filename/path")
-    #self.model: Callable = <load_your_model_here from checkpoint or folder>
-    self.model = CLIPModel.from_pretrained(os.path.join(self.base_path, "checkpoint"))
-    self.model.eval()
-    #self.text_model = CLIPTextModel.from_pretrained(os.path.join(self.base_path, "openai/clip-vit-base-patch32"))
-    self.processor = CLIPProcessor.from_pretrained(os.path.join(self.base_path, "checkpoint"))
-
-  #Add relevant model type decorator to the method below (see docs/model_types for ref.)
-  @config.inference.wrap_func
-  def get_predictions(self, input_data, **kwargs):
-    """
-    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
-    """
-    outputs = []
-    for inp in input_data:
-      image, text = inp["image"], inp["text"]
-      with torch.no_grad():
-        inputs = self.processor(text=text, images=image, return_tensors="pt", padding=True)
-        if text is not None:
-          inputs = self.processor(text=text, return_tensors="pt", padding=True)
-          embeddings = self.model.get_text_features(**inputs)
-        else:
-          inputs = self.processor(images=image, return_tensors="pt", padding=True)
-          embeddings = self.model.get_image_features(**inputs)
-      embeddings = embeddings.squeeze().cpu().numpy()
-      outputs.append(config.inference.return_type(embedding_vector=embeddings))
-
-    return outputs

clarifai/models/model_serving/examples/multimodal_embedder/clip/1/model.py

@@ -1,74 +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
-from clarifai.models.model_serving.model_config.inference_parameter import parse_req_parameters
-
-
-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_names = [inp.name for inp in self.config_msg.input]
-
-  def execute(self, requests):
-    """
-    Serve model inference requests.
-    """
-    responses = []
-
-    for request in requests:
-      parameters = request.parameters()
-      parameters = parse_req_parameters(parameters) if parameters else {}
-
-      if len(self.input_names) == 1:
-        in_batch = pb_utils.get_input_tensor_by_name(request, self.input_names[0])
-        in_batch = in_batch.as_numpy()
-        inference_response = self.inference_obj.get_predictions(in_batch, **parameters)
-      else:
-        multi_in_batch_dict = {}
-        for input_name in self.input_names:
-          in_batch = pb_utils.get_input_tensor_by_name(request, input_name)
-          in_batch = in_batch.as_numpy() if in_batch is not None else []
-          multi_in_batch_dict.update({input_name: in_batch})
-
-        inference_response = self.inference_obj.get_predictions(multi_in_batch_dict, **parameters)
-
-      responses.append(inference_response)
-
-    return responses

clarifai/models/model_serving/examples/multimodal_embedder/clip/1/test.py

@@ -1,64 +0,0 @@
-import logging
-import os
-import unittest
-
-from clarifai.models.model_serving.models.default_test import DefaultTestInferenceModel
-
-
-class CustomTestInferenceModel(DefaultTestInferenceModel):
-  """
-  Run this file to test your implementation of InferenceModel in inference.py with default tests of Triton configuration and its output values based on basic predefined inputs
-  If you want to write custom testcase or just test output value.
-  Please follow these instrucitons:
-  1. Name your test function with prefix "test" so that pytest can execute
-  2. In order to obtain output of InferenceModel, call `self.triton_get_predictions(input_data)`.
-  3. If your input is `image` and you have set custom size of it when building model repository,
-  call `self.preprocess(image)` to obtain correct resized input
-  4. Run this test by calling
-  ```bash
-  pytest ./your_triton_folder/1/test.py
-  #to see std output
-  pytest --log-cli-level=INFO  -s ./your_triton_folder/1/test.py
-  ```
-
-  ### Examples:
-  + test text-to-image output
-  ```
-  def test_text_to_image_output(self):
-    text = "Test text"
-    output = self.triton_get_predictions(text)
-    image = output.image # uint8 np.ndarray image
-    #show or save
-  ```
-  + test visual-classifier output
-  ```
-  def test_visual_classifier(self):
-    image = cv2.imread("your/local/image.jpg") # Keep in mind of format of image (BGR or RGB)
-    output = self.triton_get_predictions(image)
-    scores = output.predicted_scores # np.ndarray
-    #process scores to get class id and its score
-    logger.info(result)
-  """
-
-  # Insert your inference parameters json path here
-  # or insert a dictionary of your_parameter_name and value, e.g dict(x=1.5, y="text", c=True)
-  # or Leave it as "" if you don't have it.
-  inference_parameters = ""
-
-  ########### Initialization. Do not change it ###########
-  __test__ = True
-
-  def setUp(self) -> None:
-    logging.info("Initializing...")
-    model_type = "multimodal-embedder"  # your model type
-    self.intitialize(
-        model_type,
-        repo_version_dir=os.path.dirname(__file__),
-        is_instance_kind_gpu=True,
-        inference_parameters=self.inference_parameters)
-
-  ########################################################
-
-
-if __name__ == '__main__':
-  unittest.main()

clarifai/models/model_serving/examples/multimodal_embedder/clip/config.pbtxt

@@ -1,29 +0,0 @@
-name: "clip"
-max_batch_size: 1
-input {
-  name: "image"
-  data_type: TYPE_UINT8
-  dims: -1
-  dims: -1
-  dims: 3
-  optional: true
-}
-input {
-  name: "text"
-  data_type: TYPE_STRING
-  dims: 1
-  optional: true
-}
-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/multimodal_embedder/clip/requirements.txt

@@ -1,4 +0,0 @@
-clarifai>9.10.5
-tritonclient[all]
-transformers==4.34.1
-torch==2.0.1

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

@@ -1,12 +0,0 @@
-## Text Classification Triton Model Examples
-
-These can be used on the fly with minimal or no changes to test deploy text classification models to the Clarifai platform. See the required files section for each model below.
-
-* ### [XLM-Roberta Tweet Sentiment Classifier](./xlm-roberta/)
-
-	Required files to run tests locally:
-
-	* Download the [model checkpoint](https://huggingface.co/cardiffnlp/twitter-xlm-roberta-base-sentiment/tree/main) and store it under `xlm-roberta/1/checkpoint/`
-	```
-	huggingface-cli download cardiffnlp/twitter-xlm-roberta-base-sentiment --local-dir xlm-roberta/1/checkpoint/ --local-dir-use-symlinks False
-	```

clarifai/models/model_serving/examples/text_classification/xlm-roberta/1/inference.py

@@ -1,62 +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
-from typing import Callable
-
-import torch
-from scipy.special import softmax
-from transformers import AutoModelForSequenceClassification, AutoTokenizer
-
-from clarifai.models.model_serving.model_config import ModelTypes, get_model_config
-from clarifai.models.model_serving.models.output import ClassifierOutput
-
-config = get_model_config(ModelTypes.text_classifier)
-
-
-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.checkpoint_path: Path = os.path.join(self.base_path, "checkpoint")
-    self.model: Callable = AutoModelForSequenceClassification.from_pretrained(self.checkpoint_path)
-    self.tokenizer: Callable = AutoTokenizer.from_pretrained(self.checkpoint_path)
-    self.device = "cuda:0" if torch.cuda.is_available() else "cpu"
-
-  @config.inference.wrap_func
-  def get_predictions(self, input_data: list, **kwargs) -> list:
-    """
-    Main model inference method.
-
-    Args:
-    -----
-      input_data: A list of input data item to predict on.
-        Input data can be an image or text, etc depending on the model type.
-
-      **kwargs: your inference parameters.
-
-    Returns:
-    --------
-      List of one of the `clarifai.models.model_serving.models.output types` or `config.inference.return_type(your_output)`. Refer to the README/docs
-    """
-    outputs = []
-    for inp in input_data:
-      encoded_input = self.tokenizer(inp, return_tensors='pt')
-      output = self.model(**encoded_input)
-      scores = output[0][0].detach().numpy()
-      scores = softmax(scores)
-      outputs.append(ClassifierOutput(predicted_scores=scores))
-
-    return outputs