def _hub_model_info()

in docker_images/diffusers/app/offline.py [0:0]

• 25 lines of code
• 3 McCabe index (conditional complexity)

    def _hub_model_info(self, model_id):
        """
        This method tries to fetch locally cached model_info if any.
        If none, it requests the Hub. Useful for pre cached private models when no token is available
        """
        if self.offline_preferred:
            cache_root = os.getenv(
                "DIFFUSERS_CACHE", os.getenv("HUGGINGFACE_HUB_CACHE", "")
            )
            folder_name = file_download.repo_folder_name(
                repo_id=model_id, repo_type="model"
            )
            folder_path = os.path.join(cache_root, folder_name)
            logger.debug("Cache folder path %s", folder_path)
            filename = os.path.join(folder_path, "hub_model_info.json")
            try:
                with open(filename, "r") as f:
                    model_data = json.load(f)
            except OSError:
                logger.info(
                    "No cached model info found in file %s found for model %s. Fetching on the hub",
                    filename,
                    model_id,
                )
            else:
                model_data = hf_api.ModelInfo(**model_data)
                return model_data
        model_data = model_info(model_id, token=self.use_auth_token)
        return model_data