clarifai 11.1.4rc1__py3-none-any.whl → 11.1.5__py3-none-any.whl

clarifai/runners/models/model_upload.py

@@ -1,607 +0,0 @@
-import os
-import re
-import sys
-import tarfile
-import time
-from string import Template
-
-import yaml
-from clarifai_grpc.grpc.api import resources_pb2, service_pb2
-from clarifai_grpc.grpc.api.status import status_code_pb2
-from google.protobuf import json_format
-from rich import print
-from rich.markup import escape
-
-from clarifai.client import BaseClient
-from clarifai.runners.utils.const import (
-    AVAILABLE_PYTHON_IMAGES, AVAILABLE_TORCH_IMAGES, CONCEPTS_REQUIRED_MODEL_TYPE,
-    DEFAULT_PYTHON_VERSION, PYTHON_BUILDER_IMAGE, PYTHON_RUNTIME_IMAGE, TORCH_BASE_IMAGE)
-from clarifai.runners.utils.loader import HuggingFaceLoader
-from clarifai.urls.helper import ClarifaiUrlHelper
-from clarifai.utils.logging import logger
-
-
-def _clear_line(n: int = 1) -> None:
-  LINE_UP = '\033[1A'  # Move cursor up one line
-  LINE_CLEAR = '\x1b[2K'  # Clear the entire line
-  for _ in range(n):
-    print(LINE_UP, end=LINE_CLEAR, flush=True)
-
-
-class ModelUploader:
-
-  def __init__(self, folder: str, validate_api_ids: bool = True, download_validation_only=False):
-    """
-    :param folder: The folder containing the model.py, config.yaml, requirements.txt and
-    checkpoints.
-    :param validate_api_ids: Whether to validate the user_id and app_id in the config file. TODO(zeiler):
-    deprecate in favor of download_validation_only.
-    :param download_validation_only: Whether to skip the API config validation. Set to True if
-    just downloading a checkpoint.
-    """
-    self._client = None
-    if not validate_api_ids:  # for backwards compatibility
-      download_validation_only = True
-    self.download_validation_only = download_validation_only
-    self.folder = self._validate_folder(folder)
-    self.config = self._load_config(os.path.join(self.folder, 'config.yaml'))
-    self._validate_config()
-    self.model_proto = self._get_model_proto()
-    self.model_id = self.model_proto.id
-    self.model_version_id = None
-    self.inference_compute_info = self._get_inference_compute_info()
-    self.is_v3 = True  # Do model build for v3
-
-  def _validate_folder(self, folder):
-    if folder == ".":
-      folder = ""  # will getcwd() next which ends with /
-    if not os.path.isabs(folder):
-      folder = os.path.join(os.getcwd(), folder)
-    logger.info(f"Validating folder: {folder}")
-    if not os.path.exists(folder):
-      raise FileNotFoundError(f"Folder {folder} not found, please provide a valid folder path")
-    files = os.listdir(folder)
-    assert "config.yaml" in files, "config.yaml not found in the folder"
-    # If just downloading we don't need requirements.txt or the python code, we do need the
-    # 1/ folder to put 1/checkpoints into.
-    assert "1" in files, "Subfolder '1' not found in the folder"
-    if not self.download_validation_only:
-      assert "requirements.txt" in files, "requirements.txt not found in the folder"
-      subfolder_files = os.listdir(os.path.join(folder, '1'))
-      assert 'model.py' in subfolder_files, "model.py not found in the folder"
-    return folder
-
-  @staticmethod
-  def _load_config(config_file: str):
-    with open(config_file, 'r') as file:
-      config = yaml.safe_load(file)
-    return config
-
-  def _validate_config_checkpoints(self):
-    """
-    Validates the checkpoints section in the config file.
-    :return: loader_type the type of loader or None if no checkpoints.
-    :return: repo_id location of checkpoint.
-    :return: hf_token token to access checkpoint.
-    """
-    assert "type" in self.config.get("checkpoints"), "No loader type specified in the config file"
-    loader_type = self.config.get("checkpoints").get("type")
-    if not loader_type:
-      logger.info("No loader type specified in the config file for checkpoints")
-      return None, None, None
-    assert loader_type == "huggingface", "Only huggingface loader supported for now"
-    if loader_type == "huggingface":
-      assert "repo_id" in self.config.get("checkpoints"), "No repo_id specified in the config file"
-      repo_id = self.config.get("checkpoints").get("repo_id")
-
-      # get from config.yaml otherwise fall back to HF_TOKEN env var.
-      hf_token = self.config.get("checkpoints").get("hf_token", os.environ.get("HF_TOKEN", None))
-      return loader_type, repo_id, hf_token
-
-  def _check_app_exists(self):
-    resp = self.client.STUB.GetApp(service_pb2.GetAppRequest(user_app_id=self.client.user_app_id))
-    if resp.status.code == status_code_pb2.SUCCESS:
-      return True
-    return False
-
-  def _validate_config_model(self):
-    assert "model" in self.config, "model section not found in the config file"
-    model = self.config.get('model')
-    assert "user_id" in model, "user_id not found in the config file"
-    assert "app_id" in model, "app_id not found in the config file"
-    assert "model_type_id" in model, "model_type_id not found in the config file"
-    assert "id" in model, "model_id not found in the config file"
-    if '.' in model.get('id'):
-      logger.error(
-          "Model ID cannot contain '.', please remove it from the model_id in the config file")
-      sys.exit(1)
-
-    assert model.get('user_id') != "", "user_id cannot be empty in the config file"
-    assert model.get('app_id') != "", "app_id cannot be empty in the config file"
-    assert model.get('model_type_id') != "", "model_type_id cannot be empty in the config file"
-    assert model.get('id') != "", "model_id cannot be empty in the config file"
-
-    if not self._check_app_exists():
-      logger.error(
-          f"App {self.client.user_app_id.app_id} not found for user {self.client.user_app_id.user_id}"
-      )
-      sys.exit(1)
-
-  def _validate_config(self):
-    if not self.download_validation_only:
-      self._validate_config_model()
-
-      assert "inference_compute_info" in self.config, "inference_compute_info not found in the config file"
-
-      if self.config.get("concepts"):
-        model_type_id = self.config.get('model').get('model_type_id')
-        assert model_type_id in CONCEPTS_REQUIRED_MODEL_TYPE, f"Model type {model_type_id} not supported for concepts"
-
-    if self.config.get("checkpoints"):
-      loader_type, _, hf_token = self._validate_config_checkpoints()
-
-      if loader_type == "huggingface" and hf_token:
-        is_valid_token = HuggingFaceLoader.validate_hftoken(hf_token)
-        if not is_valid_token:
-          logger.error(
-              "Invalid Hugging Face token provided in the config file, this might cause issues with downloading the restricted model checkpoints."
-          )
-          logger.info("Continuing without Hugging Face token")
-
-  @property
-  def client(self):
-    if self._client is None:
-      assert "model" in self.config, "model info not found in the config file"
-      model = self.config.get('model')
-      assert "user_id" in model, "user_id not found in the config file"
-      assert "app_id" in model, "app_id not found in the config file"
-      # The owner of the model and the app.
-      user_id = model.get('user_id')
-      app_id = model.get('app_id')
-
-      base = os.environ.get('CLARIFAI_API_BASE', 'https://api.clarifai.com')
-
-      self._client = BaseClient(user_id=user_id, app_id=app_id, base=base)
-
-    return self._client
-
-  @property
-  def model_url(self):
-    url_helper = ClarifaiUrlHelper(self._client.auth_helper)
-    if self.model_version_id is not None:
-      return url_helper.clarifai_url(self.client.user_app_id.user_id,
-                                     self.client.user_app_id.app_id, "models", self.model_id)
-    else:
-      return url_helper.clarifai_url(self.client.user_app_id.user_id,
-                                     self.client.user_app_id.app_id, "models", self.model_id,
-                                     self.model_version_id)
-
-  def _get_model_proto(self):
-    assert "model" in self.config, "model info not found in the config file"
-    model = self.config.get('model')
-
-    assert "model_type_id" in model, "model_type_id not found in the config file"
-    assert "id" in model, "model_id not found in the config file"
-    assert "user_id" in model, "user_id not found in the config file"
-    assert "app_id" in model, "app_id not found in the config file"
-
-    model_proto = json_format.ParseDict(model, resources_pb2.Model())
-
-    return model_proto
-
-  def _get_inference_compute_info(self):
-    assert ("inference_compute_info" in self.config
-           ), "inference_compute_info not found in the config file"
-    inference_compute_info = self.config.get('inference_compute_info')
-    return json_format.ParseDict(inference_compute_info, resources_pb2.ComputeInfo())
-
-  def check_model_exists(self):
-    resp = self.client.STUB.GetModel(
-        service_pb2.GetModelRequest(
-            user_app_id=self.client.user_app_id, model_id=self.model_proto.id))
-    if resp.status.code == status_code_pb2.SUCCESS:
-      return True
-    return False
-
-  def maybe_create_model(self):
-    if self.check_model_exists():
-      logger.info(
-          f"Model '{self.client.user_app_id.user_id}/{self.client.user_app_id.app_id}/models/{self.model_proto.id}' already exists, "
-          f"will create a new version for it.")
-      return
-
-    request = service_pb2.PostModelsRequest(
-        user_app_id=self.client.user_app_id,
-        models=[self.model_proto],
-    )
-    return self.client.STUB.PostModels(request)
-
-  def _parse_requirements(self):
-    # parse the user's requirements.txt to determine the proper base image to build on top of, based on the torch and other large dependencies and it's versions
-    # List of dependencies to look for
-    dependencies = [
-        'torch',
-    ]
-    # Escape dependency names for regex
-    dep_pattern = '|'.join(map(re.escape, dependencies))
-    # All possible version specifiers
-    version_specifiers = '==|>=|<=|!=|~=|>|<'
-    # Compile a regex pattern with verbose mode for readability
-    pattern = re.compile(r"""
-          ^\s*                                   # Start of line, optional whitespace
-          (?P<dependency>""" + dep_pattern + r""")   # Dependency name
-          \s*                                   # Optional whitespace
-          (?P<specifier>""" + version_specifiers + r""")?  # Optional version specifier
-          \s*                                   # Optional whitespace
-          (?P<version>[^\s;]+)?                 # Optional version (up to space or semicolon)
-          """, re.VERBOSE)
-
-    deendencies_version = {}
-    with open(os.path.join(self.folder, 'requirements.txt'), 'r') as file:
-      for line in file:
-        # Skip empty lines and comments
-        line = line.strip()
-        if not line or line.startswith('#'):
-          continue
-        match = pattern.match(line)
-        if match:
-          dependency = match.group('dependency')
-          version = match.group('version')
-          if dependency == "torch" and line.find(
-              'whl/cpu') > 0:  # Ignore torch-cpu whl files, use base mage.
-            continue
-
-          deendencies_version[dependency] = version if version else None
-    return deendencies_version
-
-  def create_dockerfile(self):
-    dockerfile_template = os.path.join(
-        os.path.dirname(os.path.dirname(__file__)),
-        'dockerfile_template',
-        'Dockerfile.template',
-    )
-
-    with open(dockerfile_template, 'r') as template_file:
-      dockerfile_template = template_file.read()
-
-    dockerfile_template = Template(dockerfile_template)
-
-    # Get the Python version from the config file
-    build_info = self.config.get('build_info', {})
-    if 'python_version' in build_info:
-      python_version = build_info['python_version']
-      if python_version not in AVAILABLE_PYTHON_IMAGES:
-        logger.error(
-            f"Python version {python_version} not supported, please use one of the following versions: {AVAILABLE_PYTHON_IMAGES}"
-        )
-        return
-      logger.info(
-          f"Using Python version {python_version} from the config file to build the Dockerfile")
-    else:
-      logger.info(
-          f"Python version not found in the config file, using default Python version: {DEFAULT_PYTHON_VERSION}"
-      )
-      python_version = DEFAULT_PYTHON_VERSION
-
-    # This is always the final image used for runtime.
-    runtime_image = PYTHON_RUNTIME_IMAGE.format(python_version=python_version)
-    builder_image = PYTHON_BUILDER_IMAGE.format(python_version=python_version)
-    downloader_image = PYTHON_BUILDER_IMAGE.format(python_version=python_version)
-
-    # Parse the requirements.txt file to determine the base image
-    dependencies = self._parse_requirements()
-    if 'torch' in dependencies and dependencies['torch']:
-      torch_version = dependencies['torch']
-
-      # Sort in reverse so that newer cuda versions come first and are preferred.
-      for image in sorted(AVAILABLE_TORCH_IMAGES, reverse=True):
-        if torch_version in image and f'py{python_version}' in image:
-          cuda_version = image.split('-')[-1].replace('cuda', '')
-          builder_image = TORCH_BASE_IMAGE.format(
-              torch_version=torch_version,
-              python_version=python_version,
-              cuda_version=cuda_version,
-          )
-          # download_image = base_image
-          logger.info(f"Using Torch version {torch_version} base image to build the Docker image")
-          break
-    # else:  # if not torch then use the download image for the base image too
-    #   # base_image = download_image
-    #   requirements_image = base_image
-    # Replace placeholders with actual values
-    dockerfile_content = dockerfile_template.safe_substitute(
-        name='main',
-        BUILDER_IMAGE=builder_image,  # for pip requirements
-        RUNTIME_IMAGE=runtime_image,  # for runtime
-        DOWNLOADER_IMAGE=downloader_image,  # for downloading checkpoints
-    )
-
-    # Write Dockerfile
-    with open(os.path.join(self.folder, 'Dockerfile'), 'w') as dockerfile:
-      dockerfile.write(dockerfile_content)
-
-  @property
-  def checkpoint_path(self):
-    return self._checkpoint_path(self.folder)
-
-  def _checkpoint_path(self, folder):
-    return os.path.join(folder, self.checkpoint_suffix)
-
-  @property
-  def checkpoint_suffix(self):
-    return '1/checkpoints'
-
-  @property
-  def tar_file(self):
-    return f"{self.folder}.tar.gz"
-
-  def download_checkpoints(self, checkpoint_path_override: str = None):
-    """
-    Downloads the checkpoints specified in the config file.
-
-    :param checkpoint_path_override: The path to download the checkpoints to. If not provided, the
-    default path is used based on the folder ModelUploader was initialized with. The
-    checkpoint_suffix will be appended to the path.
-    """
-    if not self.config.get("checkpoints"):
-      logger.info("No checkpoints specified in the config file")
-      return True
-
-    loader_type, repo_id, hf_token = self._validate_config_checkpoints()
-
-    success = True
-    if loader_type == "huggingface":
-      loader = HuggingFaceLoader(repo_id=repo_id, token=hf_token)
-      path = self._checkpoint_path(
-          checkpoint_path_override) if checkpoint_path_override else self.checkpoint_path
-      success = loader.download_checkpoints(path)
-
-    if loader_type:
-      if not success:
-        logger.error(f"Failed to download checkpoints for model {repo_id}")
-        sys.exit(1)
-      else:
-        logger.info(f"Downloaded checkpoints for model {repo_id}")
-    return success
-
-  def _concepts_protos_from_concepts(self, concepts):
-    concept_protos = []
-    for concept in concepts:
-      concept_protos.append(resources_pb2.Concept(
-          id=str(concept[0]),
-          name=concept[1],
-      ))
-    return concept_protos
-
-  def hf_labels_to_config(self, labels, config_file):
-    with open(config_file, 'r') as file:
-      config = yaml.safe_load(file)
-    model = config.get('model')
-    model_type_id = model.get('model_type_id')
-    assert model_type_id in CONCEPTS_REQUIRED_MODEL_TYPE, f"Model type {model_type_id} not supported for concepts"
-    concept_protos = self._concepts_protos_from_concepts(labels)
-
-    config['concepts'] = [{'id': concept.id, 'name': concept.name} for concept in concept_protos]
-
-    with open(config_file, 'w') as file:
-      yaml.dump(config, file, sort_keys=False)
-    concepts = config.get('concepts')
-    logger.info(f"Updated config.yaml with {len(concepts)} concepts.")
-
-  def get_model_version_proto(self):
-
-    model_version_proto = resources_pb2.ModelVersion(
-        pretrained_model_config=resources_pb2.PretrainedModelConfig(),
-        inference_compute_info=self.inference_compute_info,
-    )
-
-    model_type_id = self.config.get('model').get('model_type_id')
-    if model_type_id in CONCEPTS_REQUIRED_MODEL_TYPE:
-
-      if 'concepts' in self.config:
-        labels = self.config.get('concepts')
-        logger.info(f"Found {len(labels)} concepts in the config file.")
-        for concept in labels:
-          concept_proto = json_format.ParseDict(concept, resources_pb2.Concept())
-          model_version_proto.output_info.data.concepts.append(concept_proto)
-      else:
-        labels = HuggingFaceLoader.fetch_labels(self.checkpoint_path)
-        logger.info(f"Found {len(labels)} concepts from the model checkpoints.")
-        # sort the concepts by id and then update the config file
-        labels = sorted(labels.items(), key=lambda x: int(x[0]))
-
-        config_file = os.path.join(self.folder, 'config.yaml')
-        try:
-          self.hf_labels_to_config(labels, config_file)
-        except Exception as e:
-          logger.error(f"Failed to update the config.yaml file with the concepts: {e}")
-
-        model_version_proto.output_info.data.concepts.extend(
-            self._concepts_protos_from_concepts(labels))
-    return model_version_proto
-
-  def upload_model_version(self, download_checkpoints):
-    file_path = f"{self.folder}.tar.gz"
-    logger.info(f"Will tar it into file: {file_path}")
-
-    model_type_id = self.config.get('model').get('model_type_id')
-
-    if (model_type_id in CONCEPTS_REQUIRED_MODEL_TYPE) and 'concepts' not in self.config:
-      logger.info(
-          f"Model type {model_type_id} requires concepts to be specified in the config.yaml file.."
-      )
-      if self.config.get("checkpoints"):
-        logger.info(
-            "Checkpoints specified in the config.yaml file, will download the HF model's config.json file to infer the concepts."
-        )
-
-        if not download_checkpoints and not HuggingFaceLoader.validate_config(
-            self.checkpoint_path):
-
-          input(
-              "Press Enter to download the HuggingFace model's config.json file to infer the concepts and continue..."
-          )
-          loader_type, repo_id, hf_token = self._validate_config_checkpoints()
-          if loader_type == "huggingface":
-            loader = HuggingFaceLoader(repo_id=repo_id, token=hf_token)
-            loader.download_config(self.checkpoint_path)
-
-      else:
-        logger.error(
-            "No checkpoints specified in the config.yaml file to infer the concepts. Please either specify the concepts directly in the config.yaml file or include a checkpoints section to download the HF model's config.json file to infer the concepts."
-        )
-        return
-
-    model_version_proto = self.get_model_version_proto()
-
-    def filter_func(tarinfo):
-      name = tarinfo.name
-      exclude = [self.tar_file, "*~"]
-      if not download_checkpoints:
-        exclude.append(self.checkpoint_suffix)
-      return None if any(name.endswith(ex) for ex in exclude) else tarinfo
-
-    with tarfile.open(self.tar_file, "w:gz") as tar:
-      tar.add(self.folder, arcname=".", filter=filter_func)
-    logger.info("Tarring complete, about to start upload.")
-
-    file_size = os.path.getsize(self.tar_file)
-    logger.info(f"Size of the tar is: {file_size} bytes")
-
-    self.maybe_create_model()
-    if not self.check_model_exists():
-      logger.error(f"Failed to create model: {self.model_proto.id}")
-      sys.exit(1)
-
-    for response in self.client.STUB.PostModelVersionsUpload(
-        self.model_version_stream_upload_iterator(model_version_proto, file_path),):
-      percent_completed = 0
-      if response.status.code == status_code_pb2.UPLOAD_IN_PROGRESS:
-        percent_completed = response.status.percent_completed
-      details = response.status.details
-
-      _clear_line()
-      print(
-          f"Status: {response.status.description}, "
-          f"Progress: {percent_completed}% - {details} ",
-          f"request_id: {response.status.req_id}",
-          end='\r',
-          flush=True)
-    logger.info("")
-    if response.status.code != status_code_pb2.MODEL_BUILDING:
-      logger.error(f"Failed to upload model version: {response}")
-      return
-    self.model_version_id = response.model_version_id
-    logger.info(f"Created Model Version ID: {self.model_version_id}")
-    logger.info(f"Full url to that version is: {self.model_url}")
-    try:
-      self.monitor_model_build()
-    finally:
-      if os.path.exists(self.tar_file):
-        logger.info(f"Cleaning up upload file: {self.tar_file}")
-        os.remove(self.tar_file)
-
-  def model_version_stream_upload_iterator(self, model_version_proto, file_path):
-    yield self.init_upload_model_version(model_version_proto, file_path)
-    with open(file_path, "rb") as f:
-      file_size = os.path.getsize(file_path)
-      chunk_size = int(127 * 1024 * 1024)  # 127MB chunk size
-      num_chunks = (file_size // chunk_size) + 1
-      logger.info("Uploading file...")
-      logger.info(f"File size: {file_size}")
-      logger.info(f"Chunk size: {chunk_size}")
-      logger.info(f"Number of chunks: {num_chunks}")
-      read_so_far = 0
-      for part_id in range(num_chunks):
-        try:
-          chunk_size = min(chunk_size, file_size - read_so_far)
-          chunk = f.read(chunk_size)
-          if not chunk:
-            break
-          read_so_far += len(chunk)
-          yield service_pb2.PostModelVersionsUploadRequest(
-              content_part=resources_pb2.UploadContentPart(
-                  data=chunk,
-                  part_number=part_id + 1,
-                  range_start=read_so_far,
-              ))
-        except Exception as e:
-          logger.exception(f"\nError uploading file: {e}")
-          break
-
-    if read_so_far == file_size:
-      logger.info("\nUpload complete!, waiting for model build...")
-
-  def init_upload_model_version(self, model_version_proto, file_path):
-    file_size = os.path.getsize(file_path)
-    logger.info(f"Uploading model version of model {self.model_proto.id}")
-    logger.info(f"Using file '{os.path.basename(file_path)}' of size: {file_size} bytes")
-    result = service_pb2.PostModelVersionsUploadRequest(
-        upload_config=service_pb2.PostModelVersionsUploadConfig(
-            user_app_id=self.client.user_app_id,
-            model_id=self.model_proto.id,
-            model_version=model_version_proto,
-            total_size=file_size,
-            is_v3=self.is_v3,
-        ))
-    return result
-
-  def get_model_build_logs(self):
-    logs_request = service_pb2.ListLogEntriesRequest(
-        log_type="builder",
-        user_app_id=self.client.user_app_id,
-        model_id=self.model_proto.id,
-        model_version_id=self.model_version_id,
-        page=1,
-        per_page=50)
-    response = self.client.STUB.ListLogEntries(logs_request)
-
-    return response
-
-  def monitor_model_build(self):
-    st = time.time()
-    seen_logs = set()  # To avoid duplicate log messages
-    while True:
-      resp = self.client.STUB.GetModelVersion(
-          service_pb2.GetModelVersionRequest(
-              user_app_id=self.client.user_app_id,
-              model_id=self.model_proto.id,
-              version_id=self.model_version_id,
-          ))
-      status_code = resp.model_version.status.code
-      if status_code == status_code_pb2.MODEL_BUILDING:
-        print(f"Model is building... (elapsed {time.time() - st:.1f}s)", end='\r', flush=True)
-
-        # Fetch and display the logs
-        logs = self.get_model_build_logs()
-        for log_entry in logs.log_entries:
-          if log_entry.url not in seen_logs:
-            seen_logs.add(log_entry.url)
-            logger.info(f"{escape(log_entry.message.strip())}")
-        time.sleep(1)
-      elif status_code == status_code_pb2.MODEL_TRAINED:
-        logger.info(f"\nModel build complete! (elapsed {time.time() - st:.1f}s)")
-        logger.info(f"Check out the model at {self.model_url} version: {self.model_version_id}")
-        return True
-      else:
-        logger.info(
-            f"\nModel build failed with status: {resp.model_version.status} and response {resp}")
-        return False
-
-
-def main(folder, download_checkpoints, skip_dockerfile):
-  uploader = ModelUploader(folder)
-  if download_checkpoints:
-    uploader.download_checkpoints()
-  if not skip_dockerfile:
-    uploader.create_dockerfile()
-  exists = uploader.check_model_exists()
-  if exists:
-    logger.info(
-        f"Model already exists at {uploader.model_url}, this upload will create a new version for it."
-    )
-  else:
-    logger.info(f"New model will be created at {uploader.model_url} with it's first version.")
-
-  input("Press Enter to continue...")
-  uploader.upload_model_version(download_checkpoints)

clarifai/runners/utils/#const.py#

@@ -1,30 +0,0 @@
-import os
-
-registry = os.environ.get('CLARIFAI_BASE_IMAGE_REGISTRY', 'public.ecr.aws/clarifai-models')
-
-GIT_SHA = "df565436eea93efb3e8d1eb558a0a46df29523ec"
-
-PYTHON_BASE_IMAGE = registry + '/python-base:{python_version}-' + GIT_SHA
-TORCH_BASE_IMAGE = registry + '/torch:{torch_version}-py{python_version}-cuda{cuda_version}-' + GIT_SHA
-
-# List of available python base images
-AVAILABLE_PYTHON_IMAGES = ['3.11', '3.12']
-
-DEFAULT_PYTHON_VERSION = 3.12
-
-# By default we download at runtime. 
-DEFAULT_DOWNLOAD_CHECKPOINT_WHEN = "runtime"
-
-# List of available torch images
-# Keep sorted by most recent cuda version.
-AVAILABLE_TORCH_IMAGES = [
-    '2.4.1-py3.11-cuda124',
-    '2.5.1-py3.11-cuda124',
-    '2.4.1-py3.12-cuda124',
-    '2.5.1-py3.12-cuda124',
-    # '2.4.1-py3.13-cuda124',
-    # '2.5.1-py3.13-cuda124',
-]
-CONCEPTS_REQUIRED_MODEL_TYPE = [
-    'visual-classifier', 'visual-detector', 'visual-segmenter', 'text-classifier'
-]