|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import importlib |
|
import os |
|
import sys |
|
from typing import TYPE_CHECKING |
|
|
|
|
|
__version__ = "0.30.2" |
|
|
|
|
|
|
|
|
|
_SUBMOD_ATTRS = { |
|
"_commit_scheduler": [ |
|
"CommitScheduler", |
|
], |
|
"_inference_endpoints": [ |
|
"InferenceEndpoint", |
|
"InferenceEndpointError", |
|
"InferenceEndpointStatus", |
|
"InferenceEndpointTimeoutError", |
|
"InferenceEndpointType", |
|
], |
|
"_login": [ |
|
"auth_list", |
|
"auth_switch", |
|
"interpreter_login", |
|
"login", |
|
"logout", |
|
"notebook_login", |
|
], |
|
"_snapshot_download": [ |
|
"snapshot_download", |
|
], |
|
"_space_api": [ |
|
"SpaceHardware", |
|
"SpaceRuntime", |
|
"SpaceStage", |
|
"SpaceStorage", |
|
"SpaceVariable", |
|
], |
|
"_tensorboard_logger": [ |
|
"HFSummaryWriter", |
|
], |
|
"_webhooks_payload": [ |
|
"WebhookPayload", |
|
"WebhookPayloadComment", |
|
"WebhookPayloadDiscussion", |
|
"WebhookPayloadDiscussionChanges", |
|
"WebhookPayloadEvent", |
|
"WebhookPayloadMovedTo", |
|
"WebhookPayloadRepo", |
|
"WebhookPayloadUrl", |
|
"WebhookPayloadWebhook", |
|
], |
|
"_webhooks_server": [ |
|
"WebhooksServer", |
|
"webhook_endpoint", |
|
], |
|
"community": [ |
|
"Discussion", |
|
"DiscussionComment", |
|
"DiscussionCommit", |
|
"DiscussionEvent", |
|
"DiscussionStatusChange", |
|
"DiscussionTitleChange", |
|
"DiscussionWithDetails", |
|
], |
|
"constants": [ |
|
"CONFIG_NAME", |
|
"FLAX_WEIGHTS_NAME", |
|
"HUGGINGFACE_CO_URL_HOME", |
|
"HUGGINGFACE_CO_URL_TEMPLATE", |
|
"PYTORCH_WEIGHTS_NAME", |
|
"REPO_TYPE_DATASET", |
|
"REPO_TYPE_MODEL", |
|
"REPO_TYPE_SPACE", |
|
"TF2_WEIGHTS_NAME", |
|
"TF_WEIGHTS_NAME", |
|
], |
|
"fastai_utils": [ |
|
"_save_pretrained_fastai", |
|
"from_pretrained_fastai", |
|
"push_to_hub_fastai", |
|
], |
|
"file_download": [ |
|
"HfFileMetadata", |
|
"_CACHED_NO_EXIST", |
|
"get_hf_file_metadata", |
|
"hf_hub_download", |
|
"hf_hub_url", |
|
"try_to_load_from_cache", |
|
], |
|
"hf_api": [ |
|
"Collection", |
|
"CollectionItem", |
|
"CommitInfo", |
|
"CommitOperation", |
|
"CommitOperationAdd", |
|
"CommitOperationCopy", |
|
"CommitOperationDelete", |
|
"DatasetInfo", |
|
"GitCommitInfo", |
|
"GitRefInfo", |
|
"GitRefs", |
|
"HfApi", |
|
"ModelInfo", |
|
"RepoUrl", |
|
"SpaceInfo", |
|
"User", |
|
"UserLikes", |
|
"WebhookInfo", |
|
"WebhookWatchedItem", |
|
"accept_access_request", |
|
"add_collection_item", |
|
"add_space_secret", |
|
"add_space_variable", |
|
"auth_check", |
|
"cancel_access_request", |
|
"change_discussion_status", |
|
"comment_discussion", |
|
"create_branch", |
|
"create_collection", |
|
"create_commit", |
|
"create_discussion", |
|
"create_inference_endpoint", |
|
"create_inference_endpoint_from_catalog", |
|
"create_pull_request", |
|
"create_repo", |
|
"create_tag", |
|
"create_webhook", |
|
"dataset_info", |
|
"delete_branch", |
|
"delete_collection", |
|
"delete_collection_item", |
|
"delete_file", |
|
"delete_folder", |
|
"delete_inference_endpoint", |
|
"delete_repo", |
|
"delete_space_secret", |
|
"delete_space_storage", |
|
"delete_space_variable", |
|
"delete_tag", |
|
"delete_webhook", |
|
"disable_webhook", |
|
"duplicate_space", |
|
"edit_discussion_comment", |
|
"enable_webhook", |
|
"file_exists", |
|
"get_collection", |
|
"get_dataset_tags", |
|
"get_discussion_details", |
|
"get_full_repo_name", |
|
"get_inference_endpoint", |
|
"get_model_tags", |
|
"get_paths_info", |
|
"get_repo_discussions", |
|
"get_safetensors_metadata", |
|
"get_space_runtime", |
|
"get_space_variables", |
|
"get_token_permission", |
|
"get_user_overview", |
|
"get_webhook", |
|
"grant_access", |
|
"list_accepted_access_requests", |
|
"list_collections", |
|
"list_datasets", |
|
"list_inference_catalog", |
|
"list_inference_endpoints", |
|
"list_lfs_files", |
|
"list_liked_repos", |
|
"list_models", |
|
"list_organization_members", |
|
"list_papers", |
|
"list_pending_access_requests", |
|
"list_rejected_access_requests", |
|
"list_repo_commits", |
|
"list_repo_files", |
|
"list_repo_likers", |
|
"list_repo_refs", |
|
"list_repo_tree", |
|
"list_spaces", |
|
"list_user_followers", |
|
"list_user_following", |
|
"list_webhooks", |
|
"merge_pull_request", |
|
"model_info", |
|
"move_repo", |
|
"paper_info", |
|
"parse_safetensors_file_metadata", |
|
"pause_inference_endpoint", |
|
"pause_space", |
|
"permanently_delete_lfs_files", |
|
"preupload_lfs_files", |
|
"reject_access_request", |
|
"rename_discussion", |
|
"repo_exists", |
|
"repo_info", |
|
"repo_type_and_id_from_hf_id", |
|
"request_space_hardware", |
|
"request_space_storage", |
|
"restart_space", |
|
"resume_inference_endpoint", |
|
"revision_exists", |
|
"run_as_future", |
|
"scale_to_zero_inference_endpoint", |
|
"set_space_sleep_time", |
|
"space_info", |
|
"super_squash_history", |
|
"unlike", |
|
"update_collection_item", |
|
"update_collection_metadata", |
|
"update_inference_endpoint", |
|
"update_repo_settings", |
|
"update_repo_visibility", |
|
"update_webhook", |
|
"upload_file", |
|
"upload_folder", |
|
"upload_large_folder", |
|
"whoami", |
|
], |
|
"hf_file_system": [ |
|
"HfFileSystem", |
|
"HfFileSystemFile", |
|
"HfFileSystemResolvedPath", |
|
"HfFileSystemStreamFile", |
|
], |
|
"hub_mixin": [ |
|
"ModelHubMixin", |
|
"PyTorchModelHubMixin", |
|
], |
|
"inference._client": [ |
|
"InferenceClient", |
|
"InferenceTimeoutError", |
|
], |
|
"inference._generated._async_client": [ |
|
"AsyncInferenceClient", |
|
], |
|
"inference._generated.types": [ |
|
"AudioClassificationInput", |
|
"AudioClassificationOutputElement", |
|
"AudioClassificationOutputTransform", |
|
"AudioClassificationParameters", |
|
"AudioToAudioInput", |
|
"AudioToAudioOutputElement", |
|
"AutomaticSpeechRecognitionEarlyStoppingEnum", |
|
"AutomaticSpeechRecognitionGenerationParameters", |
|
"AutomaticSpeechRecognitionInput", |
|
"AutomaticSpeechRecognitionOutput", |
|
"AutomaticSpeechRecognitionOutputChunk", |
|
"AutomaticSpeechRecognitionParameters", |
|
"ChatCompletionInput", |
|
"ChatCompletionInputFunctionDefinition", |
|
"ChatCompletionInputFunctionName", |
|
"ChatCompletionInputGrammarType", |
|
"ChatCompletionInputGrammarTypeType", |
|
"ChatCompletionInputMessage", |
|
"ChatCompletionInputMessageChunk", |
|
"ChatCompletionInputMessageChunkType", |
|
"ChatCompletionInputStreamOptions", |
|
"ChatCompletionInputTool", |
|
"ChatCompletionInputToolCall", |
|
"ChatCompletionInputToolChoiceClass", |
|
"ChatCompletionInputToolChoiceEnum", |
|
"ChatCompletionInputURL", |
|
"ChatCompletionOutput", |
|
"ChatCompletionOutputComplete", |
|
"ChatCompletionOutputFunctionDefinition", |
|
"ChatCompletionOutputLogprob", |
|
"ChatCompletionOutputLogprobs", |
|
"ChatCompletionOutputMessage", |
|
"ChatCompletionOutputToolCall", |
|
"ChatCompletionOutputTopLogprob", |
|
"ChatCompletionOutputUsage", |
|
"ChatCompletionStreamOutput", |
|
"ChatCompletionStreamOutputChoice", |
|
"ChatCompletionStreamOutputDelta", |
|
"ChatCompletionStreamOutputDeltaToolCall", |
|
"ChatCompletionStreamOutputFunction", |
|
"ChatCompletionStreamOutputLogprob", |
|
"ChatCompletionStreamOutputLogprobs", |
|
"ChatCompletionStreamOutputTopLogprob", |
|
"ChatCompletionStreamOutputUsage", |
|
"DepthEstimationInput", |
|
"DepthEstimationOutput", |
|
"DocumentQuestionAnsweringInput", |
|
"DocumentQuestionAnsweringInputData", |
|
"DocumentQuestionAnsweringOutputElement", |
|
"DocumentQuestionAnsweringParameters", |
|
"FeatureExtractionInput", |
|
"FeatureExtractionInputTruncationDirection", |
|
"FillMaskInput", |
|
"FillMaskOutputElement", |
|
"FillMaskParameters", |
|
"ImageClassificationInput", |
|
"ImageClassificationOutputElement", |
|
"ImageClassificationOutputTransform", |
|
"ImageClassificationParameters", |
|
"ImageSegmentationInput", |
|
"ImageSegmentationOutputElement", |
|
"ImageSegmentationParameters", |
|
"ImageSegmentationSubtask", |
|
"ImageToImageInput", |
|
"ImageToImageOutput", |
|
"ImageToImageParameters", |
|
"ImageToImageTargetSize", |
|
"ImageToTextEarlyStoppingEnum", |
|
"ImageToTextGenerationParameters", |
|
"ImageToTextInput", |
|
"ImageToTextOutput", |
|
"ImageToTextParameters", |
|
"ObjectDetectionBoundingBox", |
|
"ObjectDetectionInput", |
|
"ObjectDetectionOutputElement", |
|
"ObjectDetectionParameters", |
|
"Padding", |
|
"QuestionAnsweringInput", |
|
"QuestionAnsweringInputData", |
|
"QuestionAnsweringOutputElement", |
|
"QuestionAnsweringParameters", |
|
"SentenceSimilarityInput", |
|
"SentenceSimilarityInputData", |
|
"SummarizationInput", |
|
"SummarizationOutput", |
|
"SummarizationParameters", |
|
"SummarizationTruncationStrategy", |
|
"TableQuestionAnsweringInput", |
|
"TableQuestionAnsweringInputData", |
|
"TableQuestionAnsweringOutputElement", |
|
"TableQuestionAnsweringParameters", |
|
"Text2TextGenerationInput", |
|
"Text2TextGenerationOutput", |
|
"Text2TextGenerationParameters", |
|
"Text2TextGenerationTruncationStrategy", |
|
"TextClassificationInput", |
|
"TextClassificationOutputElement", |
|
"TextClassificationOutputTransform", |
|
"TextClassificationParameters", |
|
"TextGenerationInput", |
|
"TextGenerationInputGenerateParameters", |
|
"TextGenerationInputGrammarType", |
|
"TextGenerationOutput", |
|
"TextGenerationOutputBestOfSequence", |
|
"TextGenerationOutputDetails", |
|
"TextGenerationOutputFinishReason", |
|
"TextGenerationOutputPrefillToken", |
|
"TextGenerationOutputToken", |
|
"TextGenerationStreamOutput", |
|
"TextGenerationStreamOutputStreamDetails", |
|
"TextGenerationStreamOutputToken", |
|
"TextToAudioEarlyStoppingEnum", |
|
"TextToAudioGenerationParameters", |
|
"TextToAudioInput", |
|
"TextToAudioOutput", |
|
"TextToAudioParameters", |
|
"TextToImageInput", |
|
"TextToImageOutput", |
|
"TextToImageParameters", |
|
"TextToSpeechEarlyStoppingEnum", |
|
"TextToSpeechGenerationParameters", |
|
"TextToSpeechInput", |
|
"TextToSpeechOutput", |
|
"TextToSpeechParameters", |
|
"TextToVideoInput", |
|
"TextToVideoOutput", |
|
"TextToVideoParameters", |
|
"TokenClassificationAggregationStrategy", |
|
"TokenClassificationInput", |
|
"TokenClassificationOutputElement", |
|
"TokenClassificationParameters", |
|
"TranslationInput", |
|
"TranslationOutput", |
|
"TranslationParameters", |
|
"TranslationTruncationStrategy", |
|
"TypeEnum", |
|
"VideoClassificationInput", |
|
"VideoClassificationOutputElement", |
|
"VideoClassificationOutputTransform", |
|
"VideoClassificationParameters", |
|
"VisualQuestionAnsweringInput", |
|
"VisualQuestionAnsweringInputData", |
|
"VisualQuestionAnsweringOutputElement", |
|
"VisualQuestionAnsweringParameters", |
|
"ZeroShotClassificationInput", |
|
"ZeroShotClassificationOutputElement", |
|
"ZeroShotClassificationParameters", |
|
"ZeroShotImageClassificationInput", |
|
"ZeroShotImageClassificationOutputElement", |
|
"ZeroShotImageClassificationParameters", |
|
"ZeroShotObjectDetectionBoundingBox", |
|
"ZeroShotObjectDetectionInput", |
|
"ZeroShotObjectDetectionOutputElement", |
|
"ZeroShotObjectDetectionParameters", |
|
], |
|
"inference_api": [ |
|
"InferenceApi", |
|
], |
|
"keras_mixin": [ |
|
"KerasModelHubMixin", |
|
"from_pretrained_keras", |
|
"push_to_hub_keras", |
|
"save_pretrained_keras", |
|
], |
|
"repocard": [ |
|
"DatasetCard", |
|
"ModelCard", |
|
"RepoCard", |
|
"SpaceCard", |
|
"metadata_eval_result", |
|
"metadata_load", |
|
"metadata_save", |
|
"metadata_update", |
|
], |
|
"repocard_data": [ |
|
"CardData", |
|
"DatasetCardData", |
|
"EvalResult", |
|
"ModelCardData", |
|
"SpaceCardData", |
|
], |
|
"repository": [ |
|
"Repository", |
|
], |
|
"serialization": [ |
|
"StateDictSplit", |
|
"get_tf_storage_size", |
|
"get_torch_storage_id", |
|
"get_torch_storage_size", |
|
"load_state_dict_from_file", |
|
"load_torch_model", |
|
"save_torch_model", |
|
"save_torch_state_dict", |
|
"split_state_dict_into_shards_factory", |
|
"split_tf_state_dict_into_shards", |
|
"split_torch_state_dict_into_shards", |
|
], |
|
"serialization._dduf": [ |
|
"DDUFEntry", |
|
"export_entries_as_dduf", |
|
"export_folder_as_dduf", |
|
"read_dduf_file", |
|
], |
|
"utils": [ |
|
"CacheNotFound", |
|
"CachedFileInfo", |
|
"CachedRepoInfo", |
|
"CachedRevisionInfo", |
|
"CorruptedCacheException", |
|
"DeleteCacheStrategy", |
|
"HFCacheInfo", |
|
"HfFolder", |
|
"cached_assets_path", |
|
"configure_http_backend", |
|
"dump_environment_info", |
|
"get_session", |
|
"get_token", |
|
"logging", |
|
"scan_cache_dir", |
|
], |
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
__all__ = [ |
|
"AsyncInferenceClient", |
|
"AudioClassificationInput", |
|
"AudioClassificationOutputElement", |
|
"AudioClassificationOutputTransform", |
|
"AudioClassificationParameters", |
|
"AudioToAudioInput", |
|
"AudioToAudioOutputElement", |
|
"AutomaticSpeechRecognitionEarlyStoppingEnum", |
|
"AutomaticSpeechRecognitionGenerationParameters", |
|
"AutomaticSpeechRecognitionInput", |
|
"AutomaticSpeechRecognitionOutput", |
|
"AutomaticSpeechRecognitionOutputChunk", |
|
"AutomaticSpeechRecognitionParameters", |
|
"CONFIG_NAME", |
|
"CacheNotFound", |
|
"CachedFileInfo", |
|
"CachedRepoInfo", |
|
"CachedRevisionInfo", |
|
"CardData", |
|
"ChatCompletionInput", |
|
"ChatCompletionInputFunctionDefinition", |
|
"ChatCompletionInputFunctionName", |
|
"ChatCompletionInputGrammarType", |
|
"ChatCompletionInputGrammarTypeType", |
|
"ChatCompletionInputMessage", |
|
"ChatCompletionInputMessageChunk", |
|
"ChatCompletionInputMessageChunkType", |
|
"ChatCompletionInputStreamOptions", |
|
"ChatCompletionInputTool", |
|
"ChatCompletionInputToolCall", |
|
"ChatCompletionInputToolChoiceClass", |
|
"ChatCompletionInputToolChoiceEnum", |
|
"ChatCompletionInputURL", |
|
"ChatCompletionOutput", |
|
"ChatCompletionOutputComplete", |
|
"ChatCompletionOutputFunctionDefinition", |
|
"ChatCompletionOutputLogprob", |
|
"ChatCompletionOutputLogprobs", |
|
"ChatCompletionOutputMessage", |
|
"ChatCompletionOutputToolCall", |
|
"ChatCompletionOutputTopLogprob", |
|
"ChatCompletionOutputUsage", |
|
"ChatCompletionStreamOutput", |
|
"ChatCompletionStreamOutputChoice", |
|
"ChatCompletionStreamOutputDelta", |
|
"ChatCompletionStreamOutputDeltaToolCall", |
|
"ChatCompletionStreamOutputFunction", |
|
"ChatCompletionStreamOutputLogprob", |
|
"ChatCompletionStreamOutputLogprobs", |
|
"ChatCompletionStreamOutputTopLogprob", |
|
"ChatCompletionStreamOutputUsage", |
|
"Collection", |
|
"CollectionItem", |
|
"CommitInfo", |
|
"CommitOperation", |
|
"CommitOperationAdd", |
|
"CommitOperationCopy", |
|
"CommitOperationDelete", |
|
"CommitScheduler", |
|
"CorruptedCacheException", |
|
"DDUFEntry", |
|
"DatasetCard", |
|
"DatasetCardData", |
|
"DatasetInfo", |
|
"DeleteCacheStrategy", |
|
"DepthEstimationInput", |
|
"DepthEstimationOutput", |
|
"Discussion", |
|
"DiscussionComment", |
|
"DiscussionCommit", |
|
"DiscussionEvent", |
|
"DiscussionStatusChange", |
|
"DiscussionTitleChange", |
|
"DiscussionWithDetails", |
|
"DocumentQuestionAnsweringInput", |
|
"DocumentQuestionAnsweringInputData", |
|
"DocumentQuestionAnsweringOutputElement", |
|
"DocumentQuestionAnsweringParameters", |
|
"EvalResult", |
|
"FLAX_WEIGHTS_NAME", |
|
"FeatureExtractionInput", |
|
"FeatureExtractionInputTruncationDirection", |
|
"FillMaskInput", |
|
"FillMaskOutputElement", |
|
"FillMaskParameters", |
|
"GitCommitInfo", |
|
"GitRefInfo", |
|
"GitRefs", |
|
"HFCacheInfo", |
|
"HFSummaryWriter", |
|
"HUGGINGFACE_CO_URL_HOME", |
|
"HUGGINGFACE_CO_URL_TEMPLATE", |
|
"HfApi", |
|
"HfFileMetadata", |
|
"HfFileSystem", |
|
"HfFileSystemFile", |
|
"HfFileSystemResolvedPath", |
|
"HfFileSystemStreamFile", |
|
"HfFolder", |
|
"ImageClassificationInput", |
|
"ImageClassificationOutputElement", |
|
"ImageClassificationOutputTransform", |
|
"ImageClassificationParameters", |
|
"ImageSegmentationInput", |
|
"ImageSegmentationOutputElement", |
|
"ImageSegmentationParameters", |
|
"ImageSegmentationSubtask", |
|
"ImageToImageInput", |
|
"ImageToImageOutput", |
|
"ImageToImageParameters", |
|
"ImageToImageTargetSize", |
|
"ImageToTextEarlyStoppingEnum", |
|
"ImageToTextGenerationParameters", |
|
"ImageToTextInput", |
|
"ImageToTextOutput", |
|
"ImageToTextParameters", |
|
"InferenceApi", |
|
"InferenceClient", |
|
"InferenceEndpoint", |
|
"InferenceEndpointError", |
|
"InferenceEndpointStatus", |
|
"InferenceEndpointTimeoutError", |
|
"InferenceEndpointType", |
|
"InferenceTimeoutError", |
|
"KerasModelHubMixin", |
|
"ModelCard", |
|
"ModelCardData", |
|
"ModelHubMixin", |
|
"ModelInfo", |
|
"ObjectDetectionBoundingBox", |
|
"ObjectDetectionInput", |
|
"ObjectDetectionOutputElement", |
|
"ObjectDetectionParameters", |
|
"PYTORCH_WEIGHTS_NAME", |
|
"Padding", |
|
"PyTorchModelHubMixin", |
|
"QuestionAnsweringInput", |
|
"QuestionAnsweringInputData", |
|
"QuestionAnsweringOutputElement", |
|
"QuestionAnsweringParameters", |
|
"REPO_TYPE_DATASET", |
|
"REPO_TYPE_MODEL", |
|
"REPO_TYPE_SPACE", |
|
"RepoCard", |
|
"RepoUrl", |
|
"Repository", |
|
"SentenceSimilarityInput", |
|
"SentenceSimilarityInputData", |
|
"SpaceCard", |
|
"SpaceCardData", |
|
"SpaceHardware", |
|
"SpaceInfo", |
|
"SpaceRuntime", |
|
"SpaceStage", |
|
"SpaceStorage", |
|
"SpaceVariable", |
|
"StateDictSplit", |
|
"SummarizationInput", |
|
"SummarizationOutput", |
|
"SummarizationParameters", |
|
"SummarizationTruncationStrategy", |
|
"TF2_WEIGHTS_NAME", |
|
"TF_WEIGHTS_NAME", |
|
"TableQuestionAnsweringInput", |
|
"TableQuestionAnsweringInputData", |
|
"TableQuestionAnsweringOutputElement", |
|
"TableQuestionAnsweringParameters", |
|
"Text2TextGenerationInput", |
|
"Text2TextGenerationOutput", |
|
"Text2TextGenerationParameters", |
|
"Text2TextGenerationTruncationStrategy", |
|
"TextClassificationInput", |
|
"TextClassificationOutputElement", |
|
"TextClassificationOutputTransform", |
|
"TextClassificationParameters", |
|
"TextGenerationInput", |
|
"TextGenerationInputGenerateParameters", |
|
"TextGenerationInputGrammarType", |
|
"TextGenerationOutput", |
|
"TextGenerationOutputBestOfSequence", |
|
"TextGenerationOutputDetails", |
|
"TextGenerationOutputFinishReason", |
|
"TextGenerationOutputPrefillToken", |
|
"TextGenerationOutputToken", |
|
"TextGenerationStreamOutput", |
|
"TextGenerationStreamOutputStreamDetails", |
|
"TextGenerationStreamOutputToken", |
|
"TextToAudioEarlyStoppingEnum", |
|
"TextToAudioGenerationParameters", |
|
"TextToAudioInput", |
|
"TextToAudioOutput", |
|
"TextToAudioParameters", |
|
"TextToImageInput", |
|
"TextToImageOutput", |
|
"TextToImageParameters", |
|
"TextToSpeechEarlyStoppingEnum", |
|
"TextToSpeechGenerationParameters", |
|
"TextToSpeechInput", |
|
"TextToSpeechOutput", |
|
"TextToSpeechParameters", |
|
"TextToVideoInput", |
|
"TextToVideoOutput", |
|
"TextToVideoParameters", |
|
"TokenClassificationAggregationStrategy", |
|
"TokenClassificationInput", |
|
"TokenClassificationOutputElement", |
|
"TokenClassificationParameters", |
|
"TranslationInput", |
|
"TranslationOutput", |
|
"TranslationParameters", |
|
"TranslationTruncationStrategy", |
|
"TypeEnum", |
|
"User", |
|
"UserLikes", |
|
"VideoClassificationInput", |
|
"VideoClassificationOutputElement", |
|
"VideoClassificationOutputTransform", |
|
"VideoClassificationParameters", |
|
"VisualQuestionAnsweringInput", |
|
"VisualQuestionAnsweringInputData", |
|
"VisualQuestionAnsweringOutputElement", |
|
"VisualQuestionAnsweringParameters", |
|
"WebhookInfo", |
|
"WebhookPayload", |
|
"WebhookPayloadComment", |
|
"WebhookPayloadDiscussion", |
|
"WebhookPayloadDiscussionChanges", |
|
"WebhookPayloadEvent", |
|
"WebhookPayloadMovedTo", |
|
"WebhookPayloadRepo", |
|
"WebhookPayloadUrl", |
|
"WebhookPayloadWebhook", |
|
"WebhookWatchedItem", |
|
"WebhooksServer", |
|
"ZeroShotClassificationInput", |
|
"ZeroShotClassificationOutputElement", |
|
"ZeroShotClassificationParameters", |
|
"ZeroShotImageClassificationInput", |
|
"ZeroShotImageClassificationOutputElement", |
|
"ZeroShotImageClassificationParameters", |
|
"ZeroShotObjectDetectionBoundingBox", |
|
"ZeroShotObjectDetectionInput", |
|
"ZeroShotObjectDetectionOutputElement", |
|
"ZeroShotObjectDetectionParameters", |
|
"_CACHED_NO_EXIST", |
|
"_save_pretrained_fastai", |
|
"accept_access_request", |
|
"add_collection_item", |
|
"add_space_secret", |
|
"add_space_variable", |
|
"auth_check", |
|
"auth_list", |
|
"auth_switch", |
|
"cached_assets_path", |
|
"cancel_access_request", |
|
"change_discussion_status", |
|
"comment_discussion", |
|
"configure_http_backend", |
|
"create_branch", |
|
"create_collection", |
|
"create_commit", |
|
"create_discussion", |
|
"create_inference_endpoint", |
|
"create_inference_endpoint_from_catalog", |
|
"create_pull_request", |
|
"create_repo", |
|
"create_tag", |
|
"create_webhook", |
|
"dataset_info", |
|
"delete_branch", |
|
"delete_collection", |
|
"delete_collection_item", |
|
"delete_file", |
|
"delete_folder", |
|
"delete_inference_endpoint", |
|
"delete_repo", |
|
"delete_space_secret", |
|
"delete_space_storage", |
|
"delete_space_variable", |
|
"delete_tag", |
|
"delete_webhook", |
|
"disable_webhook", |
|
"dump_environment_info", |
|
"duplicate_space", |
|
"edit_discussion_comment", |
|
"enable_webhook", |
|
"export_entries_as_dduf", |
|
"export_folder_as_dduf", |
|
"file_exists", |
|
"from_pretrained_fastai", |
|
"from_pretrained_keras", |
|
"get_collection", |
|
"get_dataset_tags", |
|
"get_discussion_details", |
|
"get_full_repo_name", |
|
"get_hf_file_metadata", |
|
"get_inference_endpoint", |
|
"get_model_tags", |
|
"get_paths_info", |
|
"get_repo_discussions", |
|
"get_safetensors_metadata", |
|
"get_session", |
|
"get_space_runtime", |
|
"get_space_variables", |
|
"get_tf_storage_size", |
|
"get_token", |
|
"get_token_permission", |
|
"get_torch_storage_id", |
|
"get_torch_storage_size", |
|
"get_user_overview", |
|
"get_webhook", |
|
"grant_access", |
|
"hf_hub_download", |
|
"hf_hub_url", |
|
"interpreter_login", |
|
"list_accepted_access_requests", |
|
"list_collections", |
|
"list_datasets", |
|
"list_inference_catalog", |
|
"list_inference_endpoints", |
|
"list_lfs_files", |
|
"list_liked_repos", |
|
"list_models", |
|
"list_organization_members", |
|
"list_papers", |
|
"list_pending_access_requests", |
|
"list_rejected_access_requests", |
|
"list_repo_commits", |
|
"list_repo_files", |
|
"list_repo_likers", |
|
"list_repo_refs", |
|
"list_repo_tree", |
|
"list_spaces", |
|
"list_user_followers", |
|
"list_user_following", |
|
"list_webhooks", |
|
"load_state_dict_from_file", |
|
"load_torch_model", |
|
"logging", |
|
"login", |
|
"logout", |
|
"merge_pull_request", |
|
"metadata_eval_result", |
|
"metadata_load", |
|
"metadata_save", |
|
"metadata_update", |
|
"model_info", |
|
"move_repo", |
|
"notebook_login", |
|
"paper_info", |
|
"parse_safetensors_file_metadata", |
|
"pause_inference_endpoint", |
|
"pause_space", |
|
"permanently_delete_lfs_files", |
|
"preupload_lfs_files", |
|
"push_to_hub_fastai", |
|
"push_to_hub_keras", |
|
"read_dduf_file", |
|
"reject_access_request", |
|
"rename_discussion", |
|
"repo_exists", |
|
"repo_info", |
|
"repo_type_and_id_from_hf_id", |
|
"request_space_hardware", |
|
"request_space_storage", |
|
"restart_space", |
|
"resume_inference_endpoint", |
|
"revision_exists", |
|
"run_as_future", |
|
"save_pretrained_keras", |
|
"save_torch_model", |
|
"save_torch_state_dict", |
|
"scale_to_zero_inference_endpoint", |
|
"scan_cache_dir", |
|
"set_space_sleep_time", |
|
"snapshot_download", |
|
"space_info", |
|
"split_state_dict_into_shards_factory", |
|
"split_tf_state_dict_into_shards", |
|
"split_torch_state_dict_into_shards", |
|
"super_squash_history", |
|
"try_to_load_from_cache", |
|
"unlike", |
|
"update_collection_item", |
|
"update_collection_metadata", |
|
"update_inference_endpoint", |
|
"update_repo_settings", |
|
"update_repo_visibility", |
|
"update_webhook", |
|
"upload_file", |
|
"upload_folder", |
|
"upload_large_folder", |
|
"webhook_endpoint", |
|
"whoami", |
|
] |
|
|
|
|
|
def _attach(package_name, submodules=None, submod_attrs=None): |
|
"""Attach lazily loaded submodules, functions, or other attributes. |
|
|
|
Typically, modules import submodules and attributes as follows: |
|
|
|
```py |
|
import mysubmodule |
|
import anothersubmodule |
|
|
|
from .foo import someattr |
|
``` |
|
|
|
The idea is to replace a package's `__getattr__`, `__dir__`, such that all imports |
|
work exactly the way they would with normal imports, except that the import occurs |
|
upon first use. |
|
|
|
The typical way to call this function, replacing the above imports, is: |
|
|
|
```python |
|
__getattr__, __dir__ = lazy.attach( |
|
__name__, |
|
['mysubmodule', 'anothersubmodule'], |
|
{'foo': ['someattr']} |
|
) |
|
``` |
|
This functionality requires Python 3.7 or higher. |
|
|
|
Args: |
|
package_name (`str`): |
|
Typically use `__name__`. |
|
submodules (`set`): |
|
List of submodules to attach. |
|
submod_attrs (`dict`): |
|
Dictionary of submodule -> list of attributes / functions. |
|
These attributes are imported as they are used. |
|
|
|
Returns: |
|
__getattr__, __dir__, __all__ |
|
|
|
""" |
|
if submod_attrs is None: |
|
submod_attrs = {} |
|
|
|
if submodules is None: |
|
submodules = set() |
|
else: |
|
submodules = set(submodules) |
|
|
|
attr_to_modules = {attr: mod for mod, attrs in submod_attrs.items() for attr in attrs} |
|
|
|
def __getattr__(name): |
|
if name in submodules: |
|
try: |
|
return importlib.import_module(f"{package_name}.{name}") |
|
except Exception as e: |
|
print(f"Error importing {package_name}.{name}: {e}") |
|
raise |
|
elif name in attr_to_modules: |
|
submod_path = f"{package_name}.{attr_to_modules[name]}" |
|
try: |
|
submod = importlib.import_module(submod_path) |
|
except Exception as e: |
|
print(f"Error importing {submod_path}: {e}") |
|
raise |
|
attr = getattr(submod, name) |
|
|
|
|
|
|
|
|
|
if name == attr_to_modules[name]: |
|
pkg = sys.modules[package_name] |
|
pkg.__dict__[name] = attr |
|
|
|
return attr |
|
else: |
|
raise AttributeError(f"No {package_name} attribute {name}") |
|
|
|
def __dir__(): |
|
return __all__ |
|
|
|
return __getattr__, __dir__ |
|
|
|
|
|
__getattr__, __dir__ = _attach(__name__, submodules=[], submod_attrs=_SUBMOD_ATTRS) |
|
|
|
if os.environ.get("EAGER_IMPORT", ""): |
|
for attr in __all__: |
|
__getattr__(attr) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if TYPE_CHECKING: |
|
from ._commit_scheduler import CommitScheduler |
|
from ._inference_endpoints import ( |
|
InferenceEndpoint, |
|
InferenceEndpointError, |
|
InferenceEndpointStatus, |
|
InferenceEndpointTimeoutError, |
|
InferenceEndpointType, |
|
) |
|
from ._login import ( |
|
auth_list, |
|
auth_switch, |
|
interpreter_login, |
|
login, |
|
logout, |
|
notebook_login, |
|
) |
|
from ._snapshot_download import snapshot_download |
|
from ._space_api import ( |
|
SpaceHardware, |
|
SpaceRuntime, |
|
SpaceStage, |
|
SpaceStorage, |
|
SpaceVariable, |
|
) |
|
from ._tensorboard_logger import HFSummaryWriter |
|
from ._webhooks_payload import ( |
|
WebhookPayload, |
|
WebhookPayloadComment, |
|
WebhookPayloadDiscussion, |
|
WebhookPayloadDiscussionChanges, |
|
WebhookPayloadEvent, |
|
WebhookPayloadMovedTo, |
|
WebhookPayloadRepo, |
|
WebhookPayloadUrl, |
|
WebhookPayloadWebhook, |
|
) |
|
from ._webhooks_server import ( |
|
WebhooksServer, |
|
webhook_endpoint, |
|
) |
|
from .community import ( |
|
Discussion, |
|
DiscussionComment, |
|
DiscussionCommit, |
|
DiscussionEvent, |
|
DiscussionStatusChange, |
|
DiscussionTitleChange, |
|
DiscussionWithDetails, |
|
) |
|
from .constants import ( |
|
CONFIG_NAME, |
|
FLAX_WEIGHTS_NAME, |
|
HUGGINGFACE_CO_URL_HOME, |
|
HUGGINGFACE_CO_URL_TEMPLATE, |
|
PYTORCH_WEIGHTS_NAME, |
|
REPO_TYPE_DATASET, |
|
REPO_TYPE_MODEL, |
|
REPO_TYPE_SPACE, |
|
TF2_WEIGHTS_NAME, |
|
TF_WEIGHTS_NAME, |
|
) |
|
from .fastai_utils import ( |
|
_save_pretrained_fastai, |
|
from_pretrained_fastai, |
|
push_to_hub_fastai, |
|
) |
|
from .file_download import ( |
|
_CACHED_NO_EXIST, |
|
HfFileMetadata, |
|
get_hf_file_metadata, |
|
hf_hub_download, |
|
hf_hub_url, |
|
try_to_load_from_cache, |
|
) |
|
from .hf_api import ( |
|
Collection, |
|
CollectionItem, |
|
CommitInfo, |
|
CommitOperation, |
|
CommitOperationAdd, |
|
CommitOperationCopy, |
|
CommitOperationDelete, |
|
DatasetInfo, |
|
GitCommitInfo, |
|
GitRefInfo, |
|
GitRefs, |
|
HfApi, |
|
ModelInfo, |
|
RepoUrl, |
|
SpaceInfo, |
|
User, |
|
UserLikes, |
|
WebhookInfo, |
|
WebhookWatchedItem, |
|
accept_access_request, |
|
add_collection_item, |
|
add_space_secret, |
|
add_space_variable, |
|
auth_check, |
|
cancel_access_request, |
|
change_discussion_status, |
|
comment_discussion, |
|
create_branch, |
|
create_collection, |
|
create_commit, |
|
create_discussion, |
|
create_inference_endpoint, |
|
create_inference_endpoint_from_catalog, |
|
create_pull_request, |
|
create_repo, |
|
create_tag, |
|
create_webhook, |
|
dataset_info, |
|
delete_branch, |
|
delete_collection, |
|
delete_collection_item, |
|
delete_file, |
|
delete_folder, |
|
delete_inference_endpoint, |
|
delete_repo, |
|
delete_space_secret, |
|
delete_space_storage, |
|
delete_space_variable, |
|
delete_tag, |
|
delete_webhook, |
|
disable_webhook, |
|
duplicate_space, |
|
edit_discussion_comment, |
|
enable_webhook, |
|
file_exists, |
|
get_collection, |
|
get_dataset_tags, |
|
get_discussion_details, |
|
get_full_repo_name, |
|
get_inference_endpoint, |
|
get_model_tags, |
|
get_paths_info, |
|
get_repo_discussions, |
|
get_safetensors_metadata, |
|
get_space_runtime, |
|
get_space_variables, |
|
get_token_permission, |
|
get_user_overview, |
|
get_webhook, |
|
grant_access, |
|
list_accepted_access_requests, |
|
list_collections, |
|
list_datasets, |
|
list_inference_catalog, |
|
list_inference_endpoints, |
|
list_lfs_files, |
|
list_liked_repos, |
|
list_models, |
|
list_organization_members, |
|
list_papers, |
|
list_pending_access_requests, |
|
list_rejected_access_requests, |
|
list_repo_commits, |
|
list_repo_files, |
|
list_repo_likers, |
|
list_repo_refs, |
|
list_repo_tree, |
|
list_spaces, |
|
list_user_followers, |
|
list_user_following, |
|
list_webhooks, |
|
merge_pull_request, |
|
model_info, |
|
move_repo, |
|
paper_info, |
|
parse_safetensors_file_metadata, |
|
pause_inference_endpoint, |
|
pause_space, |
|
permanently_delete_lfs_files, |
|
preupload_lfs_files, |
|
reject_access_request, |
|
rename_discussion, |
|
repo_exists, |
|
repo_info, |
|
repo_type_and_id_from_hf_id, |
|
request_space_hardware, |
|
request_space_storage, |
|
restart_space, |
|
resume_inference_endpoint, |
|
revision_exists, |
|
run_as_future, |
|
scale_to_zero_inference_endpoint, |
|
set_space_sleep_time, |
|
space_info, |
|
super_squash_history, |
|
unlike, |
|
update_collection_item, |
|
update_collection_metadata, |
|
update_inference_endpoint, |
|
update_repo_settings, |
|
update_repo_visibility, |
|
update_webhook, |
|
upload_file, |
|
upload_folder, |
|
upload_large_folder, |
|
whoami, |
|
) |
|
from .hf_file_system import ( |
|
HfFileSystem, |
|
HfFileSystemFile, |
|
HfFileSystemResolvedPath, |
|
HfFileSystemStreamFile, |
|
) |
|
from .hub_mixin import ( |
|
ModelHubMixin, |
|
PyTorchModelHubMixin, |
|
) |
|
from .inference._client import ( |
|
InferenceClient, |
|
InferenceTimeoutError, |
|
) |
|
from .inference._generated._async_client import AsyncInferenceClient |
|
from .inference._generated.types import ( |
|
AudioClassificationInput, |
|
AudioClassificationOutputElement, |
|
AudioClassificationOutputTransform, |
|
AudioClassificationParameters, |
|
AudioToAudioInput, |
|
AudioToAudioOutputElement, |
|
AutomaticSpeechRecognitionEarlyStoppingEnum, |
|
AutomaticSpeechRecognitionGenerationParameters, |
|
AutomaticSpeechRecognitionInput, |
|
AutomaticSpeechRecognitionOutput, |
|
AutomaticSpeechRecognitionOutputChunk, |
|
AutomaticSpeechRecognitionParameters, |
|
ChatCompletionInput, |
|
ChatCompletionInputFunctionDefinition, |
|
ChatCompletionInputFunctionName, |
|
ChatCompletionInputGrammarType, |
|
ChatCompletionInputGrammarTypeType, |
|
ChatCompletionInputMessage, |
|
ChatCompletionInputMessageChunk, |
|
ChatCompletionInputMessageChunkType, |
|
ChatCompletionInputStreamOptions, |
|
ChatCompletionInputTool, |
|
ChatCompletionInputToolCall, |
|
ChatCompletionInputToolChoiceClass, |
|
ChatCompletionInputToolChoiceEnum, |
|
ChatCompletionInputURL, |
|
ChatCompletionOutput, |
|
ChatCompletionOutputComplete, |
|
ChatCompletionOutputFunctionDefinition, |
|
ChatCompletionOutputLogprob, |
|
ChatCompletionOutputLogprobs, |
|
ChatCompletionOutputMessage, |
|
ChatCompletionOutputToolCall, |
|
ChatCompletionOutputTopLogprob, |
|
ChatCompletionOutputUsage, |
|
ChatCompletionStreamOutput, |
|
ChatCompletionStreamOutputChoice, |
|
ChatCompletionStreamOutputDelta, |
|
ChatCompletionStreamOutputDeltaToolCall, |
|
ChatCompletionStreamOutputFunction, |
|
ChatCompletionStreamOutputLogprob, |
|
ChatCompletionStreamOutputLogprobs, |
|
ChatCompletionStreamOutputTopLogprob, |
|
ChatCompletionStreamOutputUsage, |
|
DepthEstimationInput, |
|
DepthEstimationOutput, |
|
DocumentQuestionAnsweringInput, |
|
DocumentQuestionAnsweringInputData, |
|
DocumentQuestionAnsweringOutputElement, |
|
DocumentQuestionAnsweringParameters, |
|
FeatureExtractionInput, |
|
FeatureExtractionInputTruncationDirection, |
|
FillMaskInput, |
|
FillMaskOutputElement, |
|
FillMaskParameters, |
|
ImageClassificationInput, |
|
ImageClassificationOutputElement, |
|
ImageClassificationOutputTransform, |
|
ImageClassificationParameters, |
|
ImageSegmentationInput, |
|
ImageSegmentationOutputElement, |
|
ImageSegmentationParameters, |
|
ImageSegmentationSubtask, |
|
ImageToImageInput, |
|
ImageToImageOutput, |
|
ImageToImageParameters, |
|
ImageToImageTargetSize, |
|
ImageToTextEarlyStoppingEnum, |
|
ImageToTextGenerationParameters, |
|
ImageToTextInput, |
|
ImageToTextOutput, |
|
ImageToTextParameters, |
|
ObjectDetectionBoundingBox, |
|
ObjectDetectionInput, |
|
ObjectDetectionOutputElement, |
|
ObjectDetectionParameters, |
|
Padding, |
|
QuestionAnsweringInput, |
|
QuestionAnsweringInputData, |
|
QuestionAnsweringOutputElement, |
|
QuestionAnsweringParameters, |
|
SentenceSimilarityInput, |
|
SentenceSimilarityInputData, |
|
SummarizationInput, |
|
SummarizationOutput, |
|
SummarizationParameters, |
|
SummarizationTruncationStrategy, |
|
TableQuestionAnsweringInput, |
|
TableQuestionAnsweringInputData, |
|
TableQuestionAnsweringOutputElement, |
|
TableQuestionAnsweringParameters, |
|
Text2TextGenerationInput, |
|
Text2TextGenerationOutput, |
|
Text2TextGenerationParameters, |
|
Text2TextGenerationTruncationStrategy, |
|
TextClassificationInput, |
|
TextClassificationOutputElement, |
|
TextClassificationOutputTransform, |
|
TextClassificationParameters, |
|
TextGenerationInput, |
|
TextGenerationInputGenerateParameters, |
|
TextGenerationInputGrammarType, |
|
TextGenerationOutput, |
|
TextGenerationOutputBestOfSequence, |
|
TextGenerationOutputDetails, |
|
TextGenerationOutputFinishReason, |
|
TextGenerationOutputPrefillToken, |
|
TextGenerationOutputToken, |
|
TextGenerationStreamOutput, |
|
TextGenerationStreamOutputStreamDetails, |
|
TextGenerationStreamOutputToken, |
|
TextToAudioEarlyStoppingEnum, |
|
TextToAudioGenerationParameters, |
|
TextToAudioInput, |
|
TextToAudioOutput, |
|
TextToAudioParameters, |
|
TextToImageInput, |
|
TextToImageOutput, |
|
TextToImageParameters, |
|
TextToSpeechEarlyStoppingEnum, |
|
TextToSpeechGenerationParameters, |
|
TextToSpeechInput, |
|
TextToSpeechOutput, |
|
TextToSpeechParameters, |
|
TextToVideoInput, |
|
TextToVideoOutput, |
|
TextToVideoParameters, |
|
TokenClassificationAggregationStrategy, |
|
TokenClassificationInput, |
|
TokenClassificationOutputElement, |
|
TokenClassificationParameters, |
|
TranslationInput, |
|
TranslationOutput, |
|
TranslationParameters, |
|
TranslationTruncationStrategy, |
|
TypeEnum, |
|
VideoClassificationInput, |
|
VideoClassificationOutputElement, |
|
VideoClassificationOutputTransform, |
|
VideoClassificationParameters, |
|
VisualQuestionAnsweringInput, |
|
VisualQuestionAnsweringInputData, |
|
VisualQuestionAnsweringOutputElement, |
|
VisualQuestionAnsweringParameters, |
|
ZeroShotClassificationInput, |
|
ZeroShotClassificationOutputElement, |
|
ZeroShotClassificationParameters, |
|
ZeroShotImageClassificationInput, |
|
ZeroShotImageClassificationOutputElement, |
|
ZeroShotImageClassificationParameters, |
|
ZeroShotObjectDetectionBoundingBox, |
|
ZeroShotObjectDetectionInput, |
|
ZeroShotObjectDetectionOutputElement, |
|
ZeroShotObjectDetectionParameters, |
|
) |
|
from .inference_api import InferenceApi |
|
from .keras_mixin import ( |
|
KerasModelHubMixin, |
|
from_pretrained_keras, |
|
push_to_hub_keras, |
|
save_pretrained_keras, |
|
) |
|
from .repocard import ( |
|
DatasetCard, |
|
ModelCard, |
|
RepoCard, |
|
SpaceCard, |
|
metadata_eval_result, |
|
metadata_load, |
|
metadata_save, |
|
metadata_update, |
|
) |
|
from .repocard_data import ( |
|
CardData, |
|
DatasetCardData, |
|
EvalResult, |
|
ModelCardData, |
|
SpaceCardData, |
|
) |
|
from .repository import Repository |
|
from .serialization import ( |
|
StateDictSplit, |
|
get_tf_storage_size, |
|
get_torch_storage_id, |
|
get_torch_storage_size, |
|
load_state_dict_from_file, |
|
load_torch_model, |
|
save_torch_model, |
|
save_torch_state_dict, |
|
split_state_dict_into_shards_factory, |
|
split_tf_state_dict_into_shards, |
|
split_torch_state_dict_into_shards, |
|
) |
|
from .serialization._dduf import ( |
|
DDUFEntry, |
|
export_entries_as_dduf, |
|
export_folder_as_dduf, |
|
read_dduf_file, |
|
) |
|
from .utils import ( |
|
CachedFileInfo, |
|
CachedRepoInfo, |
|
CachedRevisionInfo, |
|
CacheNotFound, |
|
CorruptedCacheException, |
|
DeleteCacheStrategy, |
|
HFCacheInfo, |
|
HfFolder, |
|
cached_assets_path, |
|
configure_http_backend, |
|
dump_environment_info, |
|
get_session, |
|
get_token, |
|
logging, |
|
scan_cache_dir, |
|
) |
|
|