Transformers documentation
Auto クラス
Auto クラス
多くの場合、from_pretrained() メソッドに渡される事前学習済みモデルの名前またはパスから、使用するアーキテクチャを推測できます。
AutoConfig、AutoModel、または AutoTokenizer をインスタンス化すると、適切なアーキテクチャクラスが自動的に作成されます。例:
model = AutoModel.from_pretrained("google-bert/bert-base-cased", device_map="auto")これにより、BertModelのモデルインスタンスが作成されます。
各タスクには対応するAutoModelクラスがあります。
Autoクラスの拡張
各 auto クラスには、カスタムクラスで拡張するためのメソッドが用意されています。例えば、カスタムモデルクラス NewModel を定義する場合は、対応する NewModelConfig も定義してください。その後、以下の例のように auto クラスに登録できます。
from transformers import AutoConfig, AutoModel
AutoConfig.register("new-model", NewModelConfig)
AutoModel.register(NewModelConfig, NewModel)その後、通常どおり auto クラスを使用できます。
NewModelConfigが PreTrainedConfig のサブクラスである場合、そのmodel_type属性が、config を登録する際に使用するキー(ここでは"new-model")と同じ値に設定されていることを確認してください。同様に、
NewModelがPreTrainedModelのサブクラスである場合、config_class属性が、モデルを登録する際に使用するクラス(ここではNewModelConfig)に設定されていることを確認してください。
AutoConfig
これは汎用の設定クラスであり、from_pretrained()クラスメソッドで生成された場合、ライブラリの設定クラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_pretrained
< source >( pretrained_model_name_or_path: str | os.PathLike[str]**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co上のモデルリポジトリにホストされている事前学習済みモデル設定のmodel idを表す文字列。
- save_pretrained() メソッドまたは save_pretrained() メソッドを使用して設定ファイルを保存した ディレクトリ へのパス。例:
./my_model_directory/ - 保存済みの設定 JSON ファイルへのパス。例:
./my_model_directory/configuration.json
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制し、キャッシュされたバージョンが存在する場合にそれを上書きするかどうかを指定します。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - return_unused_kwargs (
bool, 省略可, デフォルトはFalse) — がFalse, の場合、この関数は最終的な設定オブジェクトのみを返します。Trueの場合、この関数はTuple(config, unused_kwargs)を返します。unused_kwargsは、キーが設定属性でないキー/値のペアで構成される辞書です。つまり、configの更新に使用されず、それ以外は無視されるkwargsの部分です。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - kwargs(additional keyword arguments, optional) —
kwargs の値のうち、キーが設定属性であるものは、読み込まれた値の上書きに使用されます。キーが設定属性ではないキー/値ペアの動作は、
return_unused_kwargsキーワード引数によって制御されます。
事前学習済みモデルの設定から、ライブラリの設定クラスのいずれかをインスタンス化します。
読み込まれる config オブジェクトの model_type プロパティに基づいて、インスタンス化される設定クラスが選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_path に対するパターンマッチングにフォールバックします。
- EvollaModel — EvollaConfig(EvollaConfigモデル)
- afmoe — AfmoeConfig(AfmoeConfig モデル)
- aimv2 — Aimv2Config(Aimv2Config モデル)
- aimv2_text_model — Aimv2TextConfig (Aimv2TextConfig モデル)
- aimv2_vision_model — Aimv2VisionConfig(Aimv2VisionConfig モデル)
- albert — AlbertConfig(AlbertConfig モデル)
- align — AlignConfig(AlignConfig モデル)
- align_text_model — AlignTextConfig(AlignTextConfig モデル)
- align_vision_model — AlignVisionConfig(AlignVisionConfig モデル)
- altclip — AltCLIPConfig(AltCLIPConfig モデル)
- altclip_text_model — AltCLIPTextConfig(AltCLIPTextConfig モデル)
- altclip_vision_model — AltCLIPVisionConfig(AltCLIPVisionConfig モデル)
- apertus — ApertusConfig(ApertusConfig モデル)
- arcee — ArceeConfig(ArceeConfig モデル)
- aria — AriaConfig(AriaConfig モデル)
- aria_text — AriaTextConfig(AriaTextConfig モデル)
- audio-spectrogram-transformer — ASTConfig(ASTConfig モデル)
- audioflamingo3 — AudioFlamingo3Config(AudioFlamingo3Config モデル)
- audioflamingo3_encoder — AudioFlamingo3EncoderConfig(AudioFlamingo3EncoderConfig モデル)
- autoformer — AutoformerConfig(AutoformerConfig モデル)
- axk1 — AXK1Config(AXK1Config モデル)
- axk2 — AXK2Config(AXK2Config モデル)
- aya_vision — AyaVisionConfig(AyaVisionConfig モデル)
- bailing_hybrid — BailingHybridConfig(BailingHybridConfig モデル)
- bamba — BambaConfig(BambaConfig モデル)
- bark — BarkConfig(BarkConfigモデル)
- bart — BartConfig(BartConfig モデル)
- beam — BeamConfig(BeamConfig モデル)
- beit — BeitConfig(BeitConfig モデル)
- bert — BertConfig (BertConfig モデル)
- bert-generation — BertGenerationConfig (BertGenerationConfig モデル)
- big_bird — BigBirdConfig(BigBirdConfig モデル)
- bigbird_pegasus — BigBirdPegasusConfig(BigBirdPegasusConfig モデル)
- biogpt — BioGptConfig(BioGptConfig モデル)
- bit — BitConfig(BitConfigモデル)
- bitnet — BitNetConfig(BitNetConfigモデル)
- blenderbot — BlenderbotConfig(BlenderbotConfig モデル)
- blenderbot-small — BlenderbotSmallConfig(BlenderbotSmallConfig モデル)
- blip — BlipConfig(BlipConfig モデル)
- blip-2 — Blip2Config(Blip2Config モデル)
- blip_2_qformer — Blip2QFormerConfig(Blip2QFormerConfig モデル)
- blip_2_vision_model — Blip2VisionConfig(Blip2VisionConfig モデル)
- blip_text_model — BlipTextConfig(BlipTextConfig モデル)
- blip_vision_model — BlipVisionConfig(BlipVisionConfig モデル)
- bloom — BloomConfig(BloomConfig モデル)
- blt — BltConfig(BltConfig モデル)
- blt_global_transformer —
BltGlobalTransformerConfig(BltGlobalTransformerConfig モデル) - blt_local_decoder —
BltLocalDecoderConfig(BltLocalDecoderConfig モデル) - blt_local_encoder —
BltLocalEncoderConfig(BltLocalEncoderConfig モデル) - blt_patcher —
BltPatcherConfig(BltPatcherConfig モデル) - bridgetower — BridgeTowerConfig(BridgeTowerConfig モデル)
- bridgetower_text_model — BridgeTowerTextConfig(BridgeTowerTextConfig モデル)
- bridgetower_vision_model — BridgeTowerVisionConfig(BridgeTowerVisionConfig モデル)
- bros — BrosConfig (BrosConfig モデル)
- camembert — CamembertConfig(CamembertConfig モデル)
- canary — CanaryConfig(CanaryConfig モデル)
- canary_decoder — CanaryDecoderConfig(CanaryDecoderConfig モデル)
- canine — CanineConfig(CanineConfig モデル)
- chameleon — ChameleonConfig(ChameleonConfig モデル)
- chameleon_vqgan — ChameleonVQVAEConfig(ChameleonVQVAEConfig モデル)
- chinese_clip — ChineseCLIPConfig(ChineseCLIPConfig モデル)
- chinese_clip_text_model — ChineseCLIPTextConfig(ChineseCLIPTextConfig モデル)
- chinese_clip_vision_model — ChineseCLIPVisionConfig(ChineseCLIPVisionConfig モデル)
- chmv2 — CHMv2Config (CHMv2Config モデル)
- clap — ClapConfig(ClapConfig モデル)
- clap_audio_model — ClapAudioConfig(ClapAudioConfig モデル)
- clap_text_model — ClapTextConfig(ClapTextConfig モデル)
- clip — CLIPConfig(CLIPConfig モデル)
- clip_text_model — CLIPTextConfig(CLIPTextConfig モデル)
- clip_vision_model — CLIPVisionConfig(CLIPVisionConfig モデル)
- clipseg — CLIPSegConfig(CLIPSegConfig モデル)
- clipseg_text_model — CLIPSegTextConfig(CLIPSegTextConfig モデル)
- clipseg_vision_model — CLIPSegVisionConfig(CLIPSegVisionConfig モデル)
- clvp — ClvpConfig(ClvpConfig モデル)
- clvp_decoder — ClvpDecoderConfig(ClvpDecoderConfig モデル)
- clvp_encoder — ClvpEncoderConfig(ClvpEncoderConfig モデル)
- codegen — CodeGenConfig(CodeGenConfig モデル)
- cohere — CohereConfig(CohereConfig モデル)
- cohere2 — Cohere2Config(Cohere2Config モデル)
- cohere2_moe — Cohere2MoeConfig(Cohere2MoeConfig モデル)
- cohere2_vision — Cohere2VisionConfig(Cohere2VisionConfig モデル)
- cohere_asr — CohereAsrConfig (CohereAsrConfig モデル)
- cohere_compass — CohereCompassConfig (CohereCompassConfig モデル)
- cohere_compass_text — CohereCompassTextConfig(CohereCompassTextConfig モデル)
- cohere_compass_vision — CohereCompassVisionConfig (CohereCompassVisionConfig モデル)
- colmodernvbert — ColModernVBertConfig(ColModernVBertConfig モデル)
- colpali — ColPaliConfig(ColPaliConfig モデル)
- colqwen2 — ColQwen2Config(ColQwen2Config モデル)
- conditional_detr — ConditionalDetrConfig(ConditionalDetrConfig モデル)
- convbert — ConvBertConfig(ConvBertConfig モデル)
- convnext — ConvNextConfig(ConvNextConfig モデル)
- convnextv2 — ConvNextV2Config(ConvNextV2Config モデル)
- cosmos3_edge — Cosmos3EdgeConfig (Cosmos3EdgeConfig モデル)
- cosmos3_edge_text — Cosmos3EdgeTextConfig (Cosmos3EdgeTextConfig モデル)
- cosmos3_edge_vision — Cosmos3EdgeVisionConfig(Cosmos3EdgeVisionConfig モデル)
- cosmos3_omni — Cosmos3OmniConfig(Cosmos3OmniConfig モデル)
- cpmant — CpmAntConfig(CpmAntConfigモデル)
- csm — CsmConfig(CsmConfig モデル)
- csm_depth_decoder_model — CsmDepthDecoderConfig(CsmDepthDecoderConfig モデル)
- ctrl — CTRLConfig (CTRLConfig モデル)
- cvt — CvtConfig(CvtConfig モデル)
- cwm — CwmConfig(CwmConfig モデル)
- d_fine — DFineConfig(DFineConfig モデル)
- dab-detr — DabDetrConfig(DabDetrConfig モデル)
- dac — DacConfig(DacConfigモデル)
- data2vec-audio — Data2VecAudioConfig (Data2VecAudioConfig モデル)
- data2vec-text — Data2VecTextConfig(Data2VecTextConfig モデル)
- data2vec-vision — Data2VecVisionConfig(Data2VecVisionConfigモデル)
- dbrx — DbrxConfig(DbrxConfig モデル)
- deberta — DebertaConfig(DebertaConfig モデル)
- deberta-v2 — DebertaV2Config (DebertaV2Config モデル)
- decision_transformer — DecisionTransformerConfig(DecisionTransformerConfig モデル)
- deepseek_ocr2 — DeepseekOcr2Config(DeepseekOcr2Config モデル)
- deepseek_ocr2_encoder — DeepseekOcr2VisionEncoderConfig (DeepseekOcr2VisionEncoderConfig モデル)
- deepseek_ocr2_sam_vision_model — DeepseekOcr2SamVisionConfig(DeepseekOcr2SamVisionConfig モデル)
- deepseek_ocr2_text — DeepseekOcr2TextConfig(DeepseekOcr2TextConfig モデル)
- deepseek_ocr2_vision — DeepseekOcr2VisionConfig (DeepseekOcr2VisionConfig モデル)
- deepseek_v2 — DeepseekV2Config(DeepseekV2Config モデル)
- deepseek_v3 — DeepseekV3Config(DeepseekV3Config モデル)
- deepseek_v32 — DeepseekV32Config(DeepseekV32Config モデル)
- deepseek_v4 — DeepseekV4Config(DeepseekV4Config モデル)
- deepseek_vl — DeepseekVLConfig(DeepseekVLConfig モデル)
- deepseek_vl_hybrid — DeepseekVLHybridConfig(DeepseekVLHybridConfig モデル)
- deformable_detr — DeformableDetrConfig(DeformableDetrConfig モデル)
- deimv2 — Deimv2Config(Deimv2Config モデル)
- deit — DeiTConfig(DeiTConfig モデル)
- depth_anything — DepthAnythingConfig(DepthAnythingConfig モデル)
- depth_pro — DepthProConfig(DepthProConfig モデル)
- detr — DetrConfig(DetrConfig モデル)
- dia — DiaConfig(DiaConfigモデル)
- dia_decoder — DiaDecoderConfig(DiaDecoderConfig モデル)
- dia_encoder — DiaEncoderConfig(DiaEncoderConfig モデル)
- diffllama — DiffLlamaConfig(DiffLlamaConfig モデル)
- diffusion_gemma — DiffusionGemmaConfig(DiffusionGemmaConfig モデル)
- diffusion_gemma_text — DiffusionGemmaTextConfig(DiffusionGemmaTextConfig モデル)
- dinat — DinatConfig(DinatConfigモデル)
- dinov2 — Dinov2Config (Dinov2Config モデル)
- dinov2_with_registers — Dinov2WithRegistersConfig(Dinov2WithRegistersConfig モデル)
- dinov3_convnext — DINOv3ConvNextConfig(DINOv3ConvNextConfig モデル)
- dinov3_vit — DINOv3ViTConfig(DINOv3ViTConfig モデル)
- distilbert — DistilBertConfig(DistilBertConfigモデル)
- doge — DogeConfig (DogeConfig モデル)
- donut-swin — DonutSwinConfig(DonutSwinConfig モデル)
- dots1 — Dots1Config(Dots1Config モデル)
- dots3_note — Dots3NoteConfig(Dots3NoteConfig モデル)
- dots3_note_audio_encoder — Dots3NoteAudioConfig(Dots3NoteAudioConfig モデル)
- dots3_note_text — Dots3NoteTextConfig(Dots3NoteTextConfig モデル)
- dots3_note_vision_encoder — Dots3NoteVisionConfig(Dots3NoteVisionConfig モデル)
- dpr — DPRConfig(DPRConfig モデル)
- dpt — DPTConfig(DPTConfig モデル)
- edgetam — EdgeTamConfig(EdgeTamConfig モデル)
- edgetam_mask_decoder — EdgeTamMaskDecoderConfig(EdgeTamMaskDecoderConfig モデル)
- edgetam_prompt_encoder — EdgeTamPromptEncoderConfig(EdgeTamPromptEncoderConfigモデル)
- edgetam_video — EdgeTamVideoConfig(EdgeTamVideoConfig モデル)
- edgetam_video_mask_decoder — EdgeTamVideoMaskDecoderConfig(EdgeTamVideoMaskDecoderConfig モデル)
- edgetam_video_prompt_encoder — EdgeTamVideoPromptEncoderConfig(EdgeTamVideoPromptEncoderConfigモデル)
- edgetam_vision_model — EdgeTamVisionConfig(EdgeTamVisionConfig モデル)
- efficientloftr — EfficientLoFTRConfig(EfficientLoFTRConfig モデル)
- efficientnet — EfficientNetConfig(EfficientNetConfig モデル)
- electra — ElectraConfig(ElectraConfig モデル)
- embedding_gemma2 — EmbeddingGemma2Config(EmbeddingGemma2Config モデル)
- embedding_gemma2_text — EmbeddingGemma2TextConfig(EmbeddingGemma2TextConfig モデル)
- emu3 — Emu3Config(Emu3Config モデル)
- emu3_text_model — Emu3TextConfig(Emu3TextConfig モデル)
- emu3_vqgan — Emu3VQVAEConfig(Emu3VQVAEConfig モデル)
- encodec — EncodecConfig(EncodecConfig モデル)
- エンコーダー・デコーダー — EncoderDecoderConfig(EncoderDecoderConfig モデル)
- eomt — EomtConfig(EomtConfigモデル)
- eomt_dinov3 — EomtDinov3Config(EomtDinov3Configモデル)
- ernie — ErnieConfig(ErnieConfig モデル)
- ernie4_5 — Ernie4_5Config(Ernie4_5Config モデル)
- ernie4_5_moe — Ernie4_5_MoeConfig (Ernie4_5_MoeConfig モデル)
- ernie4_5_vl_moe — Ernie4_5_VLMoeConfig(Ernie4_5_VLMoeConfig モデル)
- ernie4_5_vl_moe_text — Ernie4_5_VLMoeTextConfig(Ernie4_5_VLMoeTextConfig モデル)
- ernie4_5_vl_moe_vision — Ernie4_5_VLMoeVisionConfig(Ernie4_5_VLMoeVisionConfig モデル)
- esm — EsmConfig(EsmConfig モデル)
- esmc — EsmcConfig(EsmcConfig モデル)
- esmfold2 — EsmFold2Config(EsmFold2Config モデル)
- eurobert — EuroBertConfig(EuroBertConfigモデル)
- evolla — EvollaConfig(EvollaConfig モデル)
- exaone4 — Exaone4Config(Exaone4Config モデル)
- exaone4_5 — Exaone4_5_Config(Exaone4_5_Config モデル)
- exaone4_5_vision — Exaone4_5_VisionConfig(Exaone4_5_VisionConfig モデル)
- exaone_moe — ExaoneMoeConfig(ExaoneMoeConfig モデル)
- falcon — FalconConfig(FalconConfig モデル)
- falcon_h1 — FalconH1Config(FalconH1Config モデル)
- falcon_mamba — FalconMambaConfig (FalconMambaConfig モデル)
- fast_vlm — FastVlmConfig(FastVlmConfig モデル)
- fastspeech2_conformer — FastSpeech2ConformerConfig(FastSpeech2ConformerConfig モデル)
- fastspeech2_conformer_hifigan — FastSpeech2ConformerHifiGanConfig(FastSpeech2ConformerHifiGanConfig モデル)
- fastspeech2_conformer_with_hifigan — FastSpeech2ConformerWithHifiGanConfig(FastSpeech2ConformerWithHifiGanConfig モデル)
- flaubert — FlaubertConfig(FlaubertConfig モデル)
- flava — FlavaConfig(FlavaConfig モデル)
- flava_image_codebook — FlavaImageCodebookConfig(FlavaImageCodebookConfig モデル)
- flava_image_model — FlavaImageConfig(FlavaImageConfig モデル)
- flava_multimodal_model — FlavaMultimodalConfig(FlavaMultimodalConfig モデル)
- flava_text_model — FlavaTextConfig(FlavaTextConfig モデル)
- flex_olmo — FlexOlmoConfig(FlexOlmoConfig モデル)
- flint — FlintConfig(FlintConfig モデル)
- florence2 — Florence2Config(Florence2Config モデル)
- florence_vision — Florence2VisionConfig(Florence2VisionConfig モデル)
- fnet — FNetConfig (FNetConfig モデル)
- focalnet — FocalNetConfig(FocalNetConfigモデル)
- fsmt — FSMTConfig(FSMTConfigモデル)
- fun_asr_nano — FunAsrNanoConfig(FunAsrNanoConfig モデル)
- fun_asr_nano_encoder — FunAsrNanoEncoderConfig(FunAsrNanoEncoderConfig モデル)
- funnel — FunnelConfig(FunnelConfigモデル)
- fuyu — FuyuConfig(FuyuConfig モデル)
- gemma — GemmaConfig(GemmaConfig モデル)
- gemma2 — Gemma2Config(Gemma2Config モデル)
- gemma3 — Gemma3Config(Gemma3Config モデル)
- gemma3_text — Gemma3TextConfig(Gemma3TextConfig モデル)
- gemma3n — Gemma3nConfig(Gemma3nConfig モデル)
- gemma3n_audio — Gemma3nAudioConfig(Gemma3nAudioConfig モデル)
- gemma3n_text — Gemma3nTextConfig(Gemma3nTextConfig モデル)
- gemma3n_vision — Gemma3nVisionConfig(Gemma3nVisionConfig モデル)
- gemma4 — Gemma4Config (Gemma4Config モデル)
- gemma4_assistant — Gemma4AssistantConfig(Gemma4AssistantConfig モデル)
- gemma4_audio — Gemma4AudioConfig(Gemma4AudioConfig モデル)
- gemma4_text — Gemma4TextConfig(Gemma4TextConfig モデル)
- gemma4_unified — Gemma4UnifiedConfig(Gemma4UnifiedConfig モデル)
- gemma4_unified_assistant — Gemma4UnifiedAssistantConfig(Gemma4UnifiedAssistantConfig モデル)
- gemma4_unified_audio — Gemma4UnifiedAudioConfig(Gemma4UnifiedAudioConfig モデル)
- gemma4_unified_text — Gemma4UnifiedTextConfig(Gemma4UnifiedTextConfig モデル)
- gemma4_unified_vision — Gemma4UnifiedVisionConfig(Gemma4UnifiedVisionConfig モデル)
- gemma4_vision — Gemma4VisionConfig(Gemma4VisionConfig モデル)
- git — GitConfig(GitConfig モデル)
- git_vision_model — GitVisionConfig(GitVisionConfig モデル)
- glm — GlmConfig(GlmConfig モデル)
- glm4 — Glm4Config(Glm4Config モデル)
- glm46v — Glm46VConfig (Glm46VConfig モデル)
- glm4_moe — Glm4MoeConfig(Glm4MoeConfig モデル)
- glm4_moe_lite — Glm4MoeLiteConfig (Glm4MoeLiteConfig モデル)
- glm4v — Glm4vConfig(Glm4vConfig モデル)
- glm4v_moe — Glm4vMoeConfig(Glm4vMoeConfig モデル)
- glm4v_moe_text — Glm4vMoeTextConfig(Glm4vMoeTextConfig モデル)
- glm4v_moe_vision — Glm4vMoeVisionConfig(Glm4vMoeVisionConfig モデル)
- glm4v_text — Glm4vTextConfig(Glm4vTextConfig モデル)
- glm4v_vision — Glm4vVisionConfig(Glm4vVisionConfig モデル)
- glm5_next — Glm5NextConfig(Glm5NextConfig モデル)
- glm5_next_text — Glm5NextTextConfig(Glm5NextTextConfig モデル)
- glm5_next_vision — Glm5NextVisionConfig(Glm5NextVisionConfig モデル)
- glm_image — GlmImageConfig(GlmImageConfig モデル)
- glm_image_text — GlmImageTextConfig(GlmImageTextConfig モデル)
- glm_image_vision — GlmImageVisionConfig(GlmImageVisionConfig モデル)
- glm_image_vqmodel — GlmImageVQVAEConfig(GlmImageVQVAEConfig モデル)
- glm_moe_dsa — GlmMoeDsaConfig(GlmMoeDsaConfig モデル)
- glm_ocr — GlmOcrConfig(GlmOcrConfig モデル)
- glm_ocr_text — GlmOcrTextConfig(GlmOcrTextConfig モデル)
- glm_ocr_vision — GlmOcrVisionConfig(GlmOcrVisionConfig モデル)
- glmasr — GlmAsrConfig(GlmAsrConfig モデル)
- glmasr_encoder — GlmAsrEncoderConfig(GlmAsrEncoderConfig モデル)
- glmga — GlmgaConfig(GlmgaConfig モデル)
- glpn — GLPNConfig(GLPNConfig モデル)
- got_ocr2 — GotOcr2Config(GotOcr2Config モデル)
- got_ocr2_vision — GotOcr2VisionConfig(GotOcr2VisionConfig モデル)
- gpt-sw3 — GPT2Config(GPT2Configモデル)
- gpt2 — GPT2Config (GPT2Config モデル)
- gpt_bigcode — GPTBigCodeConfig(GPTBigCodeConfig モデル)
- gpt_neo — GPTNeoConfig(GPTNeoConfig モデル)
- gpt_neox — GPTNeoXConfig(GPTNeoXConfig モデル)
- gpt_neox_japanese — GPTNeoXJapaneseConfig(GPTNeoXJapaneseConfig モデル)
- gpt_oss — GptOssConfig(GptOssConfig モデル)
- gptj — GPTJConfig (GPTJConfig モデル)
- gradium_phonon_tts — GradiumPhononTTSConfig(GradiumPhononTTSConfigモデル)
- granite — GraniteConfig(GraniteConfig モデル)
- granite4_vision — Granite4VisionConfig(Granite4VisionConfig モデル)
- granite4_vision_text — Granite4VisionTextConfig(Granite4VisionTextConfig モデル)
- granite_docling — GraniteDoclingConfig(GraniteDoclingConfig モデル)
- granite_docling_text — GraniteDoclingTextConfig(GraniteDoclingTextConfig モデル)
- granite_docling_vision — GraniteDoclingVisionConfig(GraniteDoclingVisionConfig モデル)
- granite_speech — GraniteSpeechConfig(GraniteSpeechConfigモデル)
- granite_speech5_ctc — GraniteSpeech5CTCConfig(GraniteSpeech5CTCConfig モデル)
- granite_speech5_encoder — GraniteSpeech5EncoderConfig(GraniteSpeech5EncoderConfig モデル)
- granite_speech_encoder — GraniteSpeechEncoderConfig(GraniteSpeechEncoderConfig モデル)
- granite_speech_plus — GraniteSpeechPlusConfig(GraniteSpeechPlusConfig モデル)
- granite_speech_plus_encoder — GraniteSpeechPlusEncoderConfig(GraniteSpeechPlusEncoderConfig モデル)
- granite_swa — GraniteSWAConfig(GraniteSWAConfig モデル)
- granitemoe — GraniteMoeConfig(GraniteMoeConfig モデル)
- granitemoe_swa — GraniteMoeSWAConfig(GraniteMoeSWAConfig モデル)
- granitemoehybrid — GraniteMoeHybridConfig(GraniteMoeHybridConfig モデル)
- granitemoeshared — GraniteMoeSharedConfig(GraniteMoeSharedConfig モデル)
- grounding-dino — GroundingDinoConfig(GroundingDinoConfig モデル)
- groupvit — GroupViTConfig (GroupViTConfig モデル)
- groupvit_text_model — GroupViTTextConfig(GroupViTTextConfig モデル)
- groupvit_vision_model — GroupViTVisionConfig(GroupViTVisionConfig モデル)
- gte — GteConfig(GteConfig モデル)
- helium — HeliumConfig(HeliumConfig モデル)
- hgnet_v2 — HGNetV2Config(HGNetV2Config モデル)
- hiera — HieraConfig(HieraConfig モデル)
- higgs_audio_v2 — HiggsAudioV2Config (HiggsAudioV2Config モデル)
- higgs_audio_v2_tokenizer — HiggsAudioV2TokenizerConfig(HiggsAudioV2TokenizerConfig モデル)
- hrm_text — HrmTextConfig(HrmTextConfig モデル)
- hubert — HubertConfig(HubertConfig モデル)
- hunyuan_v1_dense — HunYuanDenseV1Config(HunYuanDenseV1Config モデル)
- hunyuan_v1_moe — HunYuanMoEV1Config (HunYuanMoEV1Config モデル)
- hunyuan_vl — HunYuanVLConfig(HunYuanVLConfig モデル)
- hunyuan_vl_text — HunYuanVLTextConfig (HunYuanVLTextConfig モデル)
- hunyuan_vl_vision — HunYuanVLVisionConfig (HunYuanVLVisionConfig モデル)
- hy_v3 — HYV3Config(HYV3Config モデル)
- hy_v4 — HYV4Config(HYV4Config モデル)
- hyperclovax — HyperCLOVAXConfig(HyperCLOVAXConfig モデル)
- hyperclovax_vision_v2 — HyperCLOVAXVisionV2Config(HyperCLOVAXVisionV2Config モデル)
- ibert — IBertConfig(IBertConfig モデル)
- idefics — IdeficsConfig(IdeficsConfig モデル)
- idefics2 — Idefics2Config(Idefics2Config モデル)
- idefics2_perceiver — Idefics2PerceiverConfig(Idefics2PerceiverConfig モデル)
- idefics2_vision — Idefics2VisionConfig(Idefics2VisionConfig モデル)
- idefics3 — Idefics3Config(Idefics3Config モデル)
- idefics3_vision — Idefics3VisionConfig(Idefics3VisionConfig モデル)
- idefics_perciever — IdeficsPerceiverConfig(IdeficsPerceiverConfig モデル)
- idefics_vision — IdeficsVisionConfig(IdeficsVisionConfig モデル)
- ijepa — IJepaConfig(IJepaConfig モデル)
- imagegpt — ImageGPTConfig(ImageGPTConfig モデル)
- informer — InformerConfig(InformerConfig モデル)
- inkling2 — Inkling2Config(Inkling2Config モデル)
- inkling2_text — Inkling2TextConfig(Inkling2TextConfig モデル)
- inkling2_vision — Inkling2VisionConfig(Inkling2VisionConfig モデル)
- inkling_audio — InklingAudioConfig(InklingAudioConfig モデル)
- inkling_mm_model — InklingConfig(InklingConfig モデル)
- inkling_text — InklingTextConfig(InklingTextConfig モデル)
- inkling_vision — InklingVisionConfig(InklingVisionConfig モデル)
- instructblip — InstructBlipConfig(InstructBlipConfig モデル)
- instructblip_qformer — InstructBlipQFormerConfig (InstructBlipQFormerConfig モデル)
- instructblip_vision_model — InstructBlipVisionConfig(InstructBlipVisionConfig モデル)
- instructblipvideo — InstructBlipVideoConfig(InstructBlipVideoConfig モデル)
- instructblipvideo_qformer — InstructBlipVideoQFormerConfig(InstructBlipVideoQFormerConfig モデル)
- instructblipvideo_vision_model — InstructBlipVideoVisionConfig(InstructBlipVideoVisionConfig モデル)
- internlm3 — InternLM3Config(InternLM3Config モデル)
- internvl — InternVLConfig(InternVLConfig モデル)
- internvl_vision — InternVLVisionConfig(InternVLVisionConfig モデル)
- iquestcoder — IQuestCoderConfig(IQuestCoderConfig モデル)
- jais2 — Jais2Config(Jais2Config モデル)
- jamba — JambaConfig(JambaConfig モデル)
- janus — JanusConfig(JanusConfig モデル)
- janus_vision_model — JanusVisionConfig(JanusVisionConfig モデル)
- janus_vqgan — JanusVQVAEConfig(JanusVQVAEConfig モデル)
- jetmoe — JetMoeConfig(JetMoeConfig モデル)
- jev — JevConfig(JevConfigモデル)
- jina_embeddings_v3 — JinaEmbeddingsV3Config(JinaEmbeddingsV3Config モデル)
- kimi_k25 — Kimi_K25Config(Kimi_K25Config モデル)
- kimi_k25_vision — Kimi_K25VisionConfig(Kimi_K25VisionConfig モデル)
- kimi_linear — KimiLinearConfig(KimiLinearConfig モデル)
- kolibri1 — Kolibri1Config(Kolibri1Config モデル)
- kosmos-2 — Kosmos2Config(Kosmos2Config モデル)
- kosmos-2.5 — Kosmos2_5Config (Kosmos2_5Config モデル)
- kosmos_2_5_text_model — Kosmos2_5TextConfig (Kosmos2_5TextConfig モデル)
- kosmos_2_5_vision_model — Kosmos2_5VisionConfig (Kosmos2_5VisionConfig モデル)
- kosmos_2_text_model — Kosmos2TextConfig(Kosmos2TextConfig モデル)
- kosmos_2_vision_model — Kosmos2VisionConfig(Kosmos2VisionConfig モデル)
- kyutai_speech_to_text — KyutaiSpeechToTextConfig (KyutaiSpeechToTextConfig モデル)
- laguna — LagunaConfig(LagunaConfig モデル)
- lasr_ctc — LasrCTCConfig(LasrCTCConfig モデル)
- lasr_encoder — LasrEncoderConfig(LasrEncoderConfig モデル)
- layoutlm — LayoutLMConfig(LayoutLMConfig モデル)
- layoutlmv2 — LayoutLMv2Config(LayoutLMv2Config モデル)
- layoutlmv3 — LayoutLMv3Config(LayoutLMv3Config モデル)
- layoutxlm — LayoutXLMConfig(LayoutXLMConfig モデル)
- led — LEDConfig(LEDConfig モデル)
- levit — LevitConfig(LevitConfig モデル)
- lfm2 — Lfm2Config(Lfm2Config モデル)
- lfm2_moe — Lfm2MoeConfig(Lfm2MoeConfig モデル)
- lfm2_vl — Lfm2VlConfig(Lfm2VlConfig モデル)
- lfm3 — Lfm3Config(Lfm3Configモデル)
- lfm3_vl — Lfm3VlConfig(Lfm3VlConfigモデル)
- lfm3_vl_vision — Lfm3VlVisionConfig(Lfm3VlVisionConfig モデル)
- lightglue — LightGlueConfig(LightGlueConfig モデル)
- lighton_ocr — LightOnOcrConfig(LightOnOcrConfig モデル)
- lilt — LiltConfig(LiltConfig モデル)
- llama — LlamaConfig(LlamaConfig モデル)
- llama4 — Llama4Config(Llama4Config モデル)
- llama4_text — Llama4TextConfig(Llama4TextConfig モデル)
- llama4_vision_model — Llama4VisionConfig(Llama4VisionConfig モデル)
- llava — LlavaConfig(LlavaConfig モデル)
- llava_next — LlavaNextConfig(LlavaNextConfig モデル)
- llava_next_video — LlavaNextVideoConfig(LlavaNextVideoConfig モデル)
- llava_onevision — LlavaOnevisionConfig(LlavaOnevisionConfig モデル)
- longcat_flash — LongcatFlashConfig(LongcatFlashConfig モデル)
- longformer — LongformerConfig(LongformerConfig モデル)
- longt5 — LongT5Config(LongT5Config モデル)
- luke — LukeConfig(LukeConfig モデル)
- lw_detr — LwDetrConfig(LwDetrConfig モデル)
- lw_detr_vit — LwDetrViTConfig(LwDetrViTConfig モデル)
- lxmert — LxmertConfig(LxmertConfig モデル)
- m2m_100 — M2M100Config(M2M100Config モデル)
- mamba — MambaConfig(MambaConfigモデル)
- mamba2 — Mamba2Config(Mamba2Config モデル)
- marian — MarianConfig(MarianConfig モデル)
- markuplm — MarkupLMConfig(MarkupLMConfig モデル)
- mask2former — Mask2FormerConfig(Mask2FormerConfig モデル)
- maskformer — MaskFormerConfig(MaskFormerConfig モデル)
- maskformer-swin —
MaskFormerSwinConfig(MaskFormerSwinConfig モデル) - mbart — MBartConfig(MBartConfig モデル)
- megatron-bert — MegatronBertConfig(MegatronBertConfig モデル)
- mellum — MellumConfig(MellumConfig モデル)
- metaclip_2 — MetaClip2Config(MetaClip2Config モデル)
- metaclip_2_text_model — MetaClip2TextConfig(MetaClip2TextConfig モデル)
- metaclip_2_vision_model — MetaClip2VisionConfig(MetaClip2VisionConfig モデル)
- mgp-str — MgpstrConfig(MgpstrConfigモデル)
- mimi — MimiConfig(MimiConfig モデル)
- mimi_decoder — MimiDecoderConfig(MimiDecoderConfig モデル)
- mimi_encoder — MimiEncoderConfig(MimiEncoderConfig モデル)
- mimo_v2_flash — MiMoV2FlashConfig(MiMoV2FlashConfig モデル)
- minicpm3 — MiniCPM3Config(MiniCPM3Config モデル)
- minicpm_ocr — MiniCPMOcrConfig(MiniCPMOcrConfig モデル)
- minicpm_ocr_vision — MiniCPMOcrVisionConfig(MiniCPMOcrVisionConfig モデル)
- minicpmv4_6 — MiniCPMV4_6Config(MiniCPMV4_6Config モデル)
- minicpmv4_6_vision — MiniCPMV4_6VisionConfig(MiniCPMV4_6VisionConfig モデル)
- minicpmv4_7 — MiniCPMV4_7Config(MiniCPMV4_7Config モデル)
- minicpmv4_7_vision — MiniCPMV4_7VisionConfig(MiniCPMV4_7VisionConfig モデル)
- minimax — MiniMaxConfig(MiniMaxConfig モデル)
- minimax_m2 — MiniMaxM2Config(MiniMaxM2Config モデル)
- minimax_m3_vl — MiniMaxM3VLConfig(MiniMaxM3VLConfig モデル)
- minimax_m3_vl_text — MiniMaxM3VLTextConfig(MiniMaxM3VLTextConfig モデル)
- minimax_m3_vl_vision — MiniMaxM3VLVisionConfig(MiniMaxM3VLVisionConfig モデル)
- ministral — MinistralConfig(MinistralConfig モデル)
- ministral3 — Ministral3Config(Ministral3Config モデル)
- mistral — MistralConfig(MistralConfig モデル)
- mistral3 — Mistral3Config(Mistral3Config モデル)
- mistral4 — Mistral4Config(Mistral4Config モデル)
- mistral_large4 — MistralLarge4Config(MistralLarge4Config モデル)
- mistral_large4_text — MistralLarge4TextConfig(MistralLarge4TextConfig モデル)
- mixtral — MixtralConfig(MixtralConfig モデル)
- mlcd — MLCDVisionConfig(MLCDVisionConfig モデル)
- mlcd_vision_model — MLCDVisionConfig(MLCDVisionConfig モデル)
- mllama — MllamaConfig(MllamaConfig モデル)
- mllama_text_model — MllamaTextConfig(MllamaTextConfig モデル)
- mllama_vision_model — MllamaVisionConfig(MllamaVisionConfig モデル)
- mm-grounding-dino — MMGroundingDinoConfig(MMGroundingDinoConfig モデル)
- mobilebert — MobileBertConfig(MobileBertConfig モデル)
- mobilenet_v1 — MobileNetV1Config(MobileNetV1Config モデル)
- mobilenet_v2 — MobileNetV2Config(MobileNetV2Config モデル)
- mobilevit — MobileViTConfig(MobileViTConfig モデル)
- mobilevitv2 — MobileViTV2Config(MobileViTV2Config モデル)
- modernbert — ModernBertConfig (ModernBertConfig モデル)
- modernbert-decoder — ModernBertDecoderConfig(ModernBertDecoderConfig モデル)
- modernvbert — ModernVBertConfig(ModernVBertConfig モデル)
- molmo2 — Molmo2Config(Molmo2Config モデル)
- molmo2_adapter — Molmo2AdapterConfig(Molmo2AdapterConfig モデル)
- molmo2_text — Molmo2TextConfig(Molmo2TextConfig モデル)
- molmo2_vision — Molmo2VisionConfig(Molmo2VisionConfig モデル)
- moonshine — MoonshineConfig(MoonshineConfig モデル)
- moonshine_streaming — MoonshineStreamingConfig(MoonshineStreamingConfig モデル)
- moonshine_streaming_encoder — MoonshineStreamingEncoderConfig(MoonshineStreamingEncoderConfig モデル)
- moshi — MoshiConfig(MoshiConfigモデル)
- moshi_depth — MoshiDepthConfig(MoshiDepthConfig モデル)
- moss_transcribe_diarize — MossTranscribeDiarizeConfig (MossTranscribeDiarizeConfig モデル)
- moss_transcribe_diarize_audio — MossTranscribeDiarizeAudioConfig (MossTranscribeDiarizeAudioConfig モデル)
- mpnet — MPNetConfig(MPNetConfig モデル)
- mpt — MptConfig(MptConfig モデル)
- mra — MraConfig(MraConfigモデル)
- mt5 — MT5Config (MT5Config モデル)
- muse_glimmer — MuseGlimmerConfig(MuseGlimmerConfig モデル)
- muse_glimmer_assistant — MuseGlimmerAssistantConfig(MuseGlimmerAssistantConfig モデル)
- muse_glimmer_text — MuseGlimmerTextConfig(MuseGlimmerTextConfig モデル)
- muse_glimmer_vision — MuseGlimmerVisionConfig(MuseGlimmerVisionConfig モデル)
- muse_spark — MuseSparkConfig(MuseSparkConfig モデル)
- muse_spark_audio — MuseSparkAudioConfig(MuseSparkAudioConfig モデル)
- muse_spark_text — MuseSparkTextConfig(MuseSparkTextConfig モデル)
- muse_spark_vision — MuseSparkVisionConfig(MuseSparkVisionConfig モデル)
- musicflamingo — MusicFlamingoConfig(MusicFlamingoConfig モデル)
- musicgen — MusicgenConfig(MusicgenConfig モデル)
- musicgen_decoder — MusicgenDecoderConfig(MusicgenDecoderConfig モデル)
- musicgen_melody — MusicgenMelodyConfig(MusicgenMelodyConfig モデル)
- musicgen_melody_decoder — MusicgenMelodyDecoderConfig(MusicgenMelodyDecoderConfig モデル)
- mvp — MvpConfig(MvpConfigモデル)
- nanochat — NanoChatConfig(NanoChatConfig モデル)
- nemo_speechlm — Nemotron3_5TranscribeConfig(Nemotron3_5TranscribeConfig モデル)
- nemotron — NemotronConfig(NemotronConfig モデル)
- nemotron3_5_asr — Nemotron3_5AsrConfig(Nemotron3_5AsrConfig モデル)
- nemotron3_5_transcribe_asr_audio — Nemotron3_5TranscribeAsrAudioConfig(Nemotron3_5TranscribeAsrAudioConfig モデル)
- nemotron3_5_transcribe_audio — Nemotron3_5TranscribeAudioConfig(Nemotron3_5TranscribeAudioConfig モデル)
- nemotron3_5_transcribe_word_timestamp — Nemotron3_5TranscribeWordTimestampConfig(Nemotron3_5TranscribeWordTimestampConfig モデル)
- nemotron3_diarization — Nemotron3DiarizationConfig(Nemotron3DiarizationConfig モデル)
- nemotron3_diarization_audio — Nemotron3DiarizationAudioConfig(Nemotron3DiarizationAudioConfig モデル)
- nemotron_asr_streaming — NemotronAsrStreamingConfig (NemotronAsrStreamingConfig モデル)
- nemotron_asr_streaming_encoder — NemotronAsrStreamingEncoderConfig (NemotronAsrStreamingEncoderConfig モデル)
- nemotron_h — NemotronHConfig(NemotronHConfig モデル)
- nemotron_h_omni — NemotronH_Omni_Reasoning_V3_Config(NemotronH_Omni_Reasoning_V3_Config モデル)
- neomme — NeoMMEConfig(NeoMMEConfig モデル)
- neucodec — NeuCodecConfig(NeuCodecConfig モデル)
- nllb-moe — NllbMoeConfig(NllbMoeConfig モデル)
- nomic_bert — NomicBertConfig (NomicBertConfig モデル)
- nougat — NougatConfig(NougatConfigモデル)
- nystromformer — NystromformerConfig(NystromformerConfig モデル)
- olmo — OlmoConfig(OlmoConfigモデル)
- olmo2 — Olmo2Config(Olmo2Config モデル)
- olmo3 — Olmo3Config(Olmo3Config モデル)
- olmo_hybrid — OlmoHybridConfig(OlmoHybridConfig モデル)
- olmoe — OlmoeConfig(OlmoeConfig モデル)
- omdet-turbo — OmDetTurboConfig(OmDetTurboConfig モデル)
- oneformer — OneFormerConfig(OneFormerConfig モデル)
- openai-gpt — OpenAIGPTConfig(OpenAIGPTConfig モデル)
- openai_privacy_filter — OpenAIPrivacyFilterConfig(OpenAIPrivacyFilterConfig モデル)
- opt — OPTConfig(OPTConfig モデル)
- ovis2 — Ovis2Config(Ovis2Config モデル)
- ovis2_vision — Ovis2VisionConfig(Ovis2VisionConfig モデル)
- owlv2 — Owlv2Config(Owlv2Config モデル)
- owlv2_text_model — Owlv2TextConfig (Owlv2TextConfig モデル)
- owlv2_vision_model — Owlv2VisionConfig(Owlv2VisionConfig モデル)
- owlvit — OwlViTConfig (OwlViTConfig モデル)
- owlvit_text_model — OwlViTTextConfig(OwlViTTextConfig モデル)
- owlvit_vision_model — OwlViTVisionConfig (OwlViTVisionConfig モデル)
- paddleocr_vl — PaddleOCRVLConfig(PaddleOCRVLConfig モデル)
- paddleocr_vl_text — PaddleOCRTextConfig(PaddleOCRTextConfig モデル)
- paddleocr_vl_vision — PaddleOCRVisionConfig(PaddleOCRVisionConfig モデル)
- paligemma — PaliGemmaConfig(PaliGemmaConfig モデル)
- parakeet_ctc — ParakeetCTCConfig(ParakeetCTCConfig モデル)
- parakeet_encoder — ParakeetEncoderConfig(ParakeetEncoderConfig モデル)
- parakeet_rnnt — ParakeetRNNTConfig(ParakeetRNNTConfig モデル)
- parakeet_tdt — ParakeetTDTConfig(ParakeetTDTConfig モデル)
- patchtsmixer — PatchTSMixerConfig(PatchTSMixerConfig モデル)
- patchtst — PatchTSTConfig(PatchTSTConfig モデル)
- pe_audio — PeAudioConfig(PeAudioConfig モデル)
- pe_audio_encoder — PeAudioEncoderConfig(PeAudioEncoderConfig モデル)
- pe_audio_video — PeAudioVideoConfig(PeAudioVideoConfig モデル)
- pe_audio_video_encoder — PeAudioVideoEncoderConfig(PeAudioVideoEncoderConfig モデル)
- pe_video — PeVideoConfig(PeVideoConfig モデル)
- pe_video_encoder — PeVideoEncoderConfig(PeVideoEncoderConfig モデル)
- pegasus — PegasusConfig(PegasusConfig モデル)
- pegasus_x — PegasusXConfig(PegasusXConfig モデル)
- perceiver — PerceiverConfig(PerceiverConfig モデル)
- perception_lm — PerceptionLMConfig(PerceptionLMConfig モデル)
- persimmon — PersimmonConfig(PersimmonConfig モデル)
- phi — PhiConfig(PhiConfig モデル)
- phi3 — Phi3Config(Phi3Config モデル)
- phi4_multimodal — Phi4MultimodalConfig(Phi4MultimodalConfig モデル)
- phi4_multimodal_audio — Phi4MultimodalAudioConfig(Phi4MultimodalAudioConfig モデル)
- phi4_multimodal_vision — Phi4MultimodalVisionConfig(Phi4MultimodalVisionConfig モデル)
- phimoe — PhimoeConfig(PhimoeConfig モデル)
- pi0 — PI0Config(PI0Config モデル)
- pix2struct — Pix2StructConfig(Pix2StructConfig モデル)
- pix2struct_text_model — Pix2StructTextConfig(Pix2StructTextConfig モデル)
- pix2struct_vision_model — Pix2StructVisionConfig(Pix2StructVisionConfig モデル)
- pixio — PixioConfig(PixioConfigモデル)
- pixtral — PixtralVisionConfig(PixtralVisionConfig モデル)
- plbart — PLBartConfig(PLBartConfig モデル)
- pocket_tts — PocketTTSConfig(PocketTTSConfig モデル)
- pocket_tts_audio_tokenizer — PocketTTSAudioTokenizerConfig(PocketTTSAudioTokenizerConfig モデル)
- pocket_tts_audio_tokenizer_decoder — PocketTTSAudioTokenizerDecoderConfig(PocketTTSAudioTokenizerDecoderConfig モデル)
- pocket_tts_audio_tokenizer_encoder — PocketTTSAudioTokenizerEncoderConfig(PocketTTSAudioTokenizerEncoderConfig モデル)
- poolformer — PoolFormerConfig(PoolFormerConfig モデル)
- pop2piano — Pop2PianoConfig(Pop2PianoConfig モデル)
- pp_chart2table — PPChart2TableConfig(PPChart2TableConfig モデル)
- pp_chart2table_vision — PPChart2TableVisionConfig(PPChart2TableVisionConfig モデル)
- pp_doclayout_v2 — PPDocLayoutV2Config(PPDocLayoutV2Config モデル)
- pp_doclayout_v3 — PPDocLayoutV3Config(PPDocLayoutV3Config モデル)
- pp_doclayout_v4 — PPDocLayoutV4Config(PPDocLayoutV4Config モデル)
- pp_formulanet — PPFormulaNetConfig(PPFormulaNetConfig モデル)
- pp_formulanet_text — PPFormulaNetTextConfig(PPFormulaNetTextConfig モデル)
- pp_formulanet_vision — PPFormulaNetVisionConfig(PPFormulaNetVisionConfig モデル)
- pp_lcnet — PPLCNetConfig(PPLCNetConfig モデル)
- pp_lcnet_v3 — PPLCNetV3Config(PPLCNetV3Config モデル)
- pp_lcnet_v4 — PPLCNetV4Config(PPLCNetV4Config モデル)
- pp_ocrv5_mobile_det — PPOCRV5MobileDetConfig(PPOCRV5MobileDetConfig モデル)
- pp_ocrv5_mobile_rec — PPOCRV5MobileRecConfig(PPOCRV5MobileRecConfig モデル)
- pp_ocrv5_server_det — PPOCRV5ServerDetConfig(PPOCRV5ServerDetConfig モデル)
- pp_ocrv5_server_rec — PPOCRV5ServerRecConfig(PPOCRV5ServerRecConfig モデル)
- pp_ocrv6_medium_det — PPOCRV6MediumDetConfig(PPOCRV6MediumDetConfig モデル)
- pp_ocrv6_small_det — PPOCRV6SmallDetConfig(PPOCRV6SmallDetConfig モデル)
- pp_ocrv6_small_rec — PPOCRV6SmallRecConfig(PPOCRV6SmallRecConfig モデル)
- pp_ocrv6_tiny_rec — PPOCRV6TinyRecConfig(PPOCRV6TinyRecConfig モデル)
- prompt_depth_anything — PromptDepthAnythingConfig(PromptDepthAnythingConfig モデル)
- prophetnet — ProphetNetConfig(ProphetNetConfig モデル)
- pvt — PvtConfig(PvtConfig モデル)
- pvt_v2 — PvtV2Config(PvtV2Config モデル)
- qianfan_ocr — QianfanOCRConfig(QianfanOCRConfig モデル)
- qianfan_ocr_vision — QianfanOCRVisionConfig(QianfanOCRVisionConfig モデル)
- qwen2 — Qwen2Config(Qwen2Config モデル)
- qwen2_5_omni — Qwen2_5OmniConfig(Qwen2_5OmniConfig モデル)
- qwen2_5_omni_audio_encoder — Qwen2_5OmniAudioEncoderConfig(Qwen2_5OmniAudioEncoderConfig モデル)
- qwen2_5_omni_bigvgan — Qwen2_5OmniBigVGANConfig(Qwen2_5OmniBigVGANConfig モデル)
- qwen2_5_omni_dit — Qwen2_5OmniDiTConfig (Qwen2_5OmniDiTConfig モデル)
- qwen2_5_omni_talker — Qwen2_5OmniTalkerConfig (Qwen2_5OmniTalkerConfig モデル)
- qwen2_5_omni_text — Qwen2_5OmniTextConfig(Qwen2_5OmniTextConfig モデル)
- qwen2_5_omni_thinker — Qwen2_5OmniThinkerConfig(Qwen2_5OmniThinkerConfig モデル)
- qwen2_5_omni_token2wav — Qwen2_5OmniToken2WavConfig(Qwen2_5OmniToken2WavConfig モデル)
- qwen2_5_omni_vision_encoder — Qwen2_5OmniVisionEncoderConfig(Qwen2_5OmniVisionEncoderConfig モデル)
- qwen2_5_vl — Qwen2_5_VLConfig(Qwen2_5_VLConfig モデル)
- qwen2_5_vl_text — Qwen2_5_VLTextConfig(Qwen2_5_VLTextConfig モデル)
- qwen2_5_vl_vision — Qwen2_5_VLVisionConfig(Qwen2_5_VLVisionConfig モデル)
- qwen2_audio — Qwen2AudioConfig(Qwen2AudioConfig モデル)
- qwen2_audio_encoder — Qwen2AudioEncoderConfig(Qwen2AudioEncoderConfig モデル)
- qwen2_moe — Qwen2MoeConfig(Qwen2MoeConfig モデル)
- qwen2_vl — Qwen2VLConfig(Qwen2VLConfig モデル)
- qwen2_vl_text — Qwen2VLTextConfig(Qwen2VLTextConfig モデル)
- qwen2_vl_vision — Qwen2VLVisionConfig(Qwen2VLVisionConfig モデル)
- qwen3 — Qwen3Config(Qwen3Config モデル)
- qwen3_5 — Qwen3_5Config(Qwen3_5Config モデル)
- qwen3_5_moe — Qwen3_5MoeConfig(Qwen3_5MoeConfig モデル)
- qwen3_5_moe_text — Qwen3_5MoeTextConfig(Qwen3_5MoeTextConfig モデル)
- qwen3_5_moe_vision — Qwen3_5MoeVisionConfig(Qwen3_5MoeVisionConfig モデル)
- qwen3_5_text — Qwen3_5TextConfig(Qwen3_5TextConfig モデル)
- qwen3_5_vision — Qwen3_5VisionConfig(Qwen3_5VisionConfig モデル)
- qwen3_asr — Qwen3ASRConfig(Qwen3ASRConfig モデル)
- qwen3_asr_encoder — Qwen3ASREncoderConfig(Qwen3ASREncoderConfig モデル)
- qwen3_moe — Qwen3MoeConfig(Qwen3MoeConfig モデル)
- qwen3_next — Qwen3NextConfig(Qwen3NextConfig モデル)
- qwen3_omni_moe — Qwen3OmniMoeConfig(Qwen3OmniMoeConfig モデル)
- qwen3_omni_moe_audio_encoder — Qwen3OmniMoeAudioEncoderConfig(Qwen3OmniMoeAudioEncoderConfig モデル)
- qwen3_omni_moe_talker_code_predictor — Qwen3OmniMoeTalkerCodePredictorConfig(Qwen3OmniMoeTalkerCodePredictorConfig モデル)
- qwen3_omni_moe_talker_text — Qwen3OmniMoeTalkerTextConfig(Qwen3OmniMoeTalkerTextConfig モデル)
- qwen3_omni_moe_text — Qwen3OmniMoeTextConfig(Qwen3OmniMoeTextConfig モデル)
- qwen3_omni_moe_thinker — Qwen3OmniMoeThinkerConfig(Qwen3OmniMoeThinkerConfig モデル)
- qwen3_omni_moe_vision_encoder — Qwen3OmniMoeVisionEncoderConfig(Qwen3OmniMoeVisionEncoderConfig モデル)
- qwen3_vl — Qwen3VLConfig(Qwen3VLConfig モデル)
- qwen3_vl_moe — Qwen3VLMoeConfig(Qwen3VLMoeConfig モデル)
- qwen3_vl_moe_text — Qwen3VLMoeTextConfig(Qwen3VLMoeTextConfig モデル)
- qwen3_vl_moe_vision — Qwen3VLMoeVisionConfig(Qwen3VLMoeVisionConfig モデル)
- qwen3_vl_text — Qwen3VLTextConfig(Qwen3VLTextConfig モデル)
- qwen3_vl_vision — Qwen3VLVisionConfig(Qwen3VLVisionConfig モデル)
- qwen4_exp — Qwen4ExpConfig(Qwen4ExpConfig モデル)
- qwen4_exp_text — Qwen4ExpTextConfig(Qwen4ExpTextConfig モデル)
- qwen4_exp_vision — Qwen4ExpVisionConfig(Qwen4ExpVisionConfig モデル)
- radio — RadioConfig(RadioConfig モデル)
- rag — RagConfig(RagConfig モデル)
- recurrent_gemma — RecurrentGemmaConfig(RecurrentGemmaConfig モデル)
- reformer — ReformerConfig(ReformerConfig モデル)
- regnet — RegNetConfig(RegNetConfig モデル)
- rembert — RemBertConfig (RemBertConfig モデル)
- resnet — ResNetConfig(ResNetConfig モデル)
- rf_detr — RfDetrConfig(RfDetrConfig モデル)
- rf_detr_dinov2 — RfDetrDinov2Config(RfDetrDinov2Config モデル)
- roberta — RobertaConfig(RobertaConfig モデル)
- roberta-prelayernorm — RobertaPreLayerNormConfig(RobertaPreLayerNormConfigモデル)
- roc_bert — RoCBertConfig(RoCBertConfig モデル)
- roformer — RoFormerConfig(RoFormerConfig モデル)
- rt_detr — RTDetrConfig(RTDetrConfig モデル)
- rt_detr_resnet — RTDetrResNetConfig(RTDetrResNetConfig モデル)
- rt_detr_v2 — RTDetrV2Config(RTDetrV2Config モデル)
- rwkv — RwkvConfig(RwkvConfig モデル)
- sam — SamConfig(SamConfig モデル)
- sam2 — Sam2Config(Sam2Config モデル)
- sam2_hiera_det_model — Sam2HieraDetConfig(Sam2HieraDetConfig モデル)
- sam2_mask_decoder — Sam2MaskDecoderConfig(Sam2MaskDecoderConfig モデル)
- sam2_prompt_encoder — Sam2PromptEncoderConfig(Sam2PromptEncoderConfig モデル)
- sam2_video — Sam2VideoConfig (Sam2VideoConfig モデル)
- sam2_video_mask_decoder — Sam2VideoMaskDecoderConfig(Sam2VideoMaskDecoderConfig モデル)
- sam2_video_prompt_encoder — Sam2VideoPromptEncoderConfig(Sam2VideoPromptEncoderConfig モデル)
- sam2_vision_model — Sam2VisionConfig(Sam2VisionConfig モデル)
- sam3 — Sam3Config(Sam3Config モデル)
- sam3_detr_decoder — Sam3DETRDecoderConfig(Sam3DETRDecoderConfig モデル)
- sam3_detr_encoder — Sam3DETREncoderConfig(Sam3DETREncoderConfig モデル)
- sam3_geometry_encoder — Sam3GeometryEncoderConfig(Sam3GeometryEncoderConfig モデル)
- sam3_lite_text — Sam3LiteTextConfig(Sam3LiteTextConfig モデル)
- sam3_lite_text_detr_decoder — Sam3LiteTextDETRDecoderConfig(Sam3LiteTextDETRDecoderConfig モデル)
- sam3_lite_text_detr_encoder — Sam3LiteTextDETREncoderConfig(Sam3LiteTextDETREncoderConfig モデル)
- sam3_lite_text_geometry_encoder — Sam3LiteTextGeometryEncoderConfig(Sam3LiteTextGeometryEncoderConfig モデル)
- sam3_lite_text_mask_decoder — Sam3LiteTextMaskDecoderConfig(Sam3LiteTextMaskDecoderConfig モデル)
- sam3_lite_text_text_model — Sam3LiteTextTextConfig(Sam3LiteTextTextConfig モデル)
- sam3_mask_decoder — Sam3MaskDecoderConfig(Sam3MaskDecoderConfig モデル)
- sam3_tracker — Sam3TrackerConfig(Sam3TrackerConfig モデル)
- sam3_tracker_mask_decoder — Sam3TrackerMaskDecoderConfig(Sam3TrackerMaskDecoderConfig モデル)
- sam3_tracker_prompt_encoder — Sam3TrackerPromptEncoderConfig(Sam3TrackerPromptEncoderConfig モデル)
- sam3_tracker_video — Sam3TrackerVideoConfig(Sam3TrackerVideoConfig モデル)
- sam3_tracker_video_mask_decoder — Sam3TrackerVideoMaskDecoderConfig(Sam3TrackerVideoMaskDecoderConfig モデル)
- sam3_tracker_video_prompt_encoder — Sam3TrackerVideoPromptEncoderConfig(Sam3TrackerVideoPromptEncoderConfig モデル)
- sam3_video — Sam3VideoConfig (Sam3VideoConfig モデル)
- sam3_vision_model — Sam3VisionConfig (Sam3VisionConfig モデル)
- sam3_vit_model — Sam3ViTConfig(Sam3ViTConfig モデル)
- sam_hq — SamHQConfig(SamHQConfig モデル)
- sam_hq_mask_decoder — SamHQMaskDecoderConfig(SamHQMaskDecoderConfig モデル)
- sam_hq_prompt_encoder — SamHQPromptEncoderConfig(SamHQPromptEncoderConfig モデル)
- sam_hq_vision_model — SamHQVisionConfig(SamHQVisionConfig モデル)
- sam_mask_decoder — SamMaskDecoderConfig(SamMaskDecoderConfig モデル)
- sam_prompt_encoder — SamPromptEncoderConfig(SamPromptEncoderConfig モデル)
- sam_vision_model — SamVisionConfig(SamVisionConfig モデル)
- sapiens2 — Sapiens2Config(Sapiens2Config モデル)
- sapiens2_head — Sapiens2HeadConfig(Sapiens2HeadConfig モデル)
- seamless_m4t — SeamlessM4TConfig(SeamlessM4TConfig モデル)
- seamless_m4t_v2 — SeamlessM4Tv2Config(SeamlessM4Tv2Config モデル)
- seed_oss — SeedOssConfig(SeedOssConfig モデル)
- segformer — SegformerConfig(SegformerConfig モデル)
- seggpt — SegGptConfig(SegGptConfig モデル)
- sew — SEWConfig(SEWConfig モデル)
- sew-d — SEWDConfig(SEWDConfigモデル)
- shieldgemma2 — ShieldGemma2Config(ShieldGemma2Config モデル)
- siglip — SiglipConfig(SiglipConfig モデル)
- siglip2 — Siglip2Config(Siglip2Config モデル)
- siglip2_text_model — Siglip2TextConfig(Siglip2TextConfig モデル)
- siglip2_vision_model — Siglip2VisionConfig(Siglip2VisionConfig モデル)
- siglip_text_model — SiglipTextConfig(SiglipTextConfig モデル)
- siglip_vision_model — SiglipVisionConfig(SiglipVisionConfig モデル)
- slanet — SLANetConfig(SLANetConfig モデル)
- slanext — SLANeXtConfig(SLANeXtConfig モデル)
- slanext_vision — SLANeXtVisionConfig(SLANeXtVisionConfig モデル)
- smollm3 — SmolLM3Config(SmolLM3Config モデル)
- smolvlm — SmolVLMConfig(SmolVLMConfig モデル)
- smolvlm_vision — SmolVLMVisionConfig(SmolVLMVisionConfig モデル)
- solar_open — SolarOpenConfig(SolarOpenConfig モデル)
- speech-encoder-decoder — SpeechEncoderDecoderConfig(SpeechEncoderDecoderConfig モデル)
- speech_to_text — Speech2TextConfig(Speech2TextConfig モデル)
- speecht5 — SpeechT5Config(SpeechT5Config モデル)
- speecht5_hifigan — SpeechT5HifiGanConfig(SpeechT5HifiGanConfig モデル)
- splinter — SplinterConfig(SplinterConfig モデル)
- squeezebert — SqueezeBertConfig (SqueezeBertConfig モデル)
- stablelm — StableLmConfig (StableLmConfig モデル)
- starcoder2 — Starcoder2Config (Starcoder2Config モデル)
- step3p5 — Step3p7TextConfig(Step3p7TextConfig モデル)
- step3p5_vision — Step3p7VisionConfig(Step3p7VisionConfig モデル)
- step3p7 — Step3p7Config (Step3p7Config モデル)
- superglue — SuperGlueConfig(SuperGlueConfig モデル)
- superpoint — SuperPointConfig(SuperPointConfig モデル)
- swiftformer — SwiftFormerConfig(SwiftFormerConfig モデル)
- swin — SwinConfig(SwinConfig モデル)
- swin2sr — Swin2SRConfig(Swin2SRConfig モデル)
- swinv2 — Swinv2Config(Swinv2Config モデル)
- switch_transformers — SwitchTransformersConfig(SwitchTransformersConfig モデル)
- t5 — T5Config(T5Config モデル)
- t5_gemma_module — T5GemmaModuleConfig(T5GemmaModuleConfig モデル)
- t5gemma — T5GemmaConfig(T5GemmaConfig モデル)
- t5gemma2 — T5Gemma2Config(T5Gemma2Config モデル)
- t5gemma2_decoder — T5Gemma2DecoderConfig(T5Gemma2DecoderConfig モデル)
- t5gemma2_encoder — T5Gemma2EncoderConfig(T5Gemma2EncoderConfig モデル)
- t5gemma2_text — T5Gemma2TextConfig(T5Gemma2TextConfig モデル)
- table-transformer — TableTransformerConfig(TableTransformerConfigモデル)
- tapas — TapasConfig(TapasConfig モデル)
- textnet — TextNetConfig(TextNetConfig モデル)
- time_series_transformer — TimeSeriesTransformerConfig(TimeSeriesTransformerConfig モデル)
- timesfm — TimesFmConfig (TimesFmConfig モデル)
- timesfm2_5 — TimesFm2_5Config(TimesFm2_5Config モデル)
- timesformer — TimesformerConfig(TimesformerConfig モデル)
- timm_backbone — TimmBackboneConfig(TimmBackboneConfig モデル)
- timm_wrapper — TimmWrapperConfig(TimmWrapperConfig モデル)
- tipsv2 — Tipsv2Config(Tipsv2Config モデル)
- tipsv2_dpt — Tipsv2DptConfig (Tipsv2DptConfig モデル)
- tipsv2_text_model — Tipsv2TextConfig(Tipsv2TextConfig モデル)
- tipsv2_vision_model — Tipsv2VisionConfig(Tipsv2VisionConfig モデル)
- trocr — TrOCRConfig(TrOCRConfig モデル)
- tvp — TvpConfig(TvpConfig モデル)
- udop — UdopConfig (UdopConfig モデル)
- ultrabert — UltrabertConfig(UltrabertConfig モデル)
- ultravbert — UltraVBertConfig(UltraVBertConfig モデル)
- umt5 — UMT5Config (UMT5Config モデル)
- unispeech — UniSpeechConfig(UniSpeechConfig モデル)
- unispeech-sat — UniSpeechSatConfig(UniSpeechSatConfig モデル)
- univnet — UnivNetConfig(UnivNetConfig モデル)
- unlimited_ocr — UnlimitedOcrConfig(UnlimitedOcrConfig モデル)
- unlimited_ocr_sam_vision_model — UnlimitedOcrSamVisionConfig(UnlimitedOcrSamVisionConfig モデル)
- unlimited_ocr_text — UnlimitedOcrTextConfig(UnlimitedOcrTextConfig モデル)
- unlimited_ocr_vision — UnlimitedOcrVisionConfig(UnlimitedOcrVisionConfig モデル)
- unlimited_ocr_vision_encoder — UnlimitedOcrVisionEncoderConfig(UnlimitedOcrVisionEncoderConfig モデル)
- upernet — UperNetConfig(UperNetConfig モデル)
- uvdoc — UVDocConfig(UVDocConfig モデル)
- uvdoc_backbone — UVDocBackboneConfig(UVDocBackboneConfig モデル)
- vaultgemma — VaultGemmaConfig(VaultGemmaConfig モデル)
- vibevoice — VibeVoiceConfig(VibeVoiceConfig モデル)
- vibevoice_acoustic_tokenizer — VibeVoiceAcousticTokenizerConfig(VibeVoiceAcousticTokenizerConfig モデル)
- vibevoice_acoustic_tokenizer_decoder — VibeVoiceAcousticTokenizerDecoderConfig(VibeVoiceAcousticTokenizerDecoderConfig モデル)
- vibevoice_acoustic_tokenizer_encoder — VibeVoiceAcousticTokenizerEncoderConfig(VibeVoiceAcousticTokenizerEncoderConfig モデル)
- vibevoice_asr — VibeVoiceAsrConfig(VibeVoiceAsrConfig モデル)
- vibevoice_diffusion — VibeVoiceDiffusionHeadConfig(VibeVoiceDiffusionHeadConfig モデル)
- video_llama_3 — VideoLlama3Config(VideoLlama3Config モデル)
- video_llama_3_vision — VideoLlama3VisionConfig(VideoLlama3VisionConfig モデル)
- video_llava — VideoLlavaConfig(VideoLlavaConfig モデル)
- videomae — VideoMAEConfig(VideoMAEConfig モデル)
- videomt — VideomtConfig(VideomtConfig モデル)
- videoprism — VideoPrismConfig (VideoPrismConfig モデル)
- videoprism_text_model — VideoPrismTextConfig(VideoPrismTextConfig モデル)
- videoprism_vision_model — VideoPrismVisionConfig(VideoPrismVisionConfig モデル)
- vilt — ViltConfig(ViltConfig モデル)
- vipllava — VipLlavaConfig(VipLlavaConfig モデル)
- vision-encoder-decoder — VisionEncoderDecoderConfig(VisionEncoderDecoderConfig モデル)
- vision-text-dual-encoder — VisionTextDualEncoderConfig(VisionTextDualEncoderConfig モデル)
- visual_bert — VisualBertConfig(VisualBertConfig モデル)
- vit — ViTConfig (ViTConfig モデル)
- vit_mae — ViTMAEConfig(ViTMAEConfig モデル)
- vit_msn — ViTMSNConfig(ViTMSNConfig モデル)
- vitdet — VitDetConfig(VitDetConfig モデル)
- vitmatte — VitMatteConfig(VitMatteConfig モデル)
- vitpose — VitPoseConfig (VitPoseConfig モデル)
- vitpose_backbone —
VitPoseBackboneConfig(VitPoseBackboneConfig モデル) - vits — VitsConfig(VitsConfig モデル)
- vivit — VivitConfig(VivitConfig モデル)
- vjepa2 — VJEPA2Config(VJEPA2Config モデル)
- voxtral — VoxtralConfig(VoxtralConfig モデル)
- voxtral_encoder — VoxtralEncoderConfig(VoxtralEncoderConfig モデル)
- voxtral_realtime — VoxtralRealtimeConfig(VoxtralRealtimeConfig モデル)
- voxtral_realtime_encoder — VoxtralRealtimeEncoderConfig (VoxtralRealtimeEncoderConfig モデル)
- voxtral_realtime_text — VoxtralRealtimeTextConfig(VoxtralRealtimeTextConfig モデル)
- wav2vec2 — Wav2Vec2Config (Wav2Vec2Config モデル)
- wav2vec2-bert — Wav2Vec2BertConfig(Wav2Vec2BertConfig モデル)
- wav2vec2-conformer — Wav2Vec2ConformerConfig (Wav2Vec2ConformerConfig モデル)
- wavlm — WavLMConfig(WavLMConfig モデル)
- whisper — WhisperConfig(WhisperConfig モデル)
- xclip — XCLIPConfig(XCLIPConfigモデル)
- xclip_text_model — XCLIPTextConfig(XCLIPTextConfig モデル)
- xclip_vision_model — XCLIPVisionConfig(XCLIPVisionConfig モデル)
- xcodec — XcodecConfig(XcodecConfigモデル)
- xcodec2 — Xcodec2Config(Xcodec2Config モデル)
- xglm — XGLMConfig(XGLMConfig モデル)
- xlm — XLMConfig(XLMConfig モデル)
- xlm-roberta — XLMRobertaConfig(XLMRobertaConfig モデル)
- xlm-roberta-xl — XLMRobertaXLConfig(XLMRobertaXLConfig モデル)
- xlnet — XLNetConfig(XLNetConfig モデル)
- xlstm — xLSTMConfig(xLSTMConfig モデル)
- xmod — XmodConfig(XmodConfig モデル)
- yolos — YolosConfig(YolosConfig モデル)
- yoso — YosoConfig(YosoConfigモデル)
- youtu — YoutuConfig(YoutuConfig モデル)
- zamba — ZambaConfig(ZambaConfig モデル)
- zamba2 — Zamba2Config(Zamba2Config モデル)
- zaya — ZayaConfig(ZayaConfig モデル)
- zoedepth — ZoeDepthConfig(ZoeDepthConfig モデル)
例:
>>> from transformers import AutoConfig
>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-uncased")
>>> # Download configuration from huggingface.co (user-uploaded) and cache.
>>> config = AutoConfig.from_pretrained("dbmdz/bert-base-german-cased")
>>> # If configuration file is in a directory (e.g., was saved using *save_pretrained('./test/saved_model/')*).
>>> config = AutoConfig.from_pretrained("./test/bert_saved_model/")
>>> # Load a specific configuration file.
>>> config = AutoConfig.from_pretrained("./test/bert_saved_model/my_configuration.json")
>>> # Change some config attributes when loading a pretrained config.
>>> config = AutoConfig.from_pretrained("google-bert/bert-base-uncased", output_attentions=True, foo=False)
>>> config.output_attentions
True
>>> config, unused_kwargs = AutoConfig.from_pretrained(
... "google-bert/bert-base-uncased", output_attentions=True, foo=False, return_unused_kwargs=True
... )
>>> config.output_attentions
True
>>> unused_kwargs
{'foo': False}register
< source >( model_typeconfigexist_ok = False )
パラメータ
- model_type (
str) — “bert” や “gpt” のようなモデルタイプ。 - config (PreTrainedConfig) — 登録する設定。
このクラス用の新しい設定を登録します。
AutoTokenizer
これは汎用的なトークナイザークラスであり、AutoTokenizer.from_pretrained()クラスメソッドで生成された場合、ライブラリのトークナイザークラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_pretrained
< source >( pretrained_model_name_or_path*inputs**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co上のモデルリポジトリにホストされている事前定義されたトークナイザーのmodel idを表す文字列。
- トークナイザーに必要とされる語彙ファイルを含むディレクトリへのパス。例えば、save_pretrained()メソッドを使用して保存されたものなど、
./my_model_directory/のようなパスを指定します。 - トークナイザーが単一の語彙ファイルのみを必要とする場合(Bert や XLNet など)に限り、保存された単一の語彙ファイルへのパスを指定します。例:
./my_model_directory/vocab.txt。(すべての派生クラスに適用されるわけではありません)
- inputs(追加の位置引数、任意)—
Tokenizer の
__init__()メソッドに渡されます。 - config (PreTrainedConfig、任意) — インスタンス化されるトークナイザークラスを決定するために使用される設定オブジェクトです。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制し、キャッシュされたバージョンが存在する場合にそれを上書きするかどうかを指定します。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - subfolder (
str, 省略可) — huggingface.co 上のモデルリポジトリのサブフォルダ内に該当するファイルが配置されている場合(例: facebook/rag-token-base)に、ここで指定します。 - tokenizer_type (
str, 省略可) — 読み込む Tokenizer の種類。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - kwargs(追加のキーワード引数、任意)—
Tokenizer の
__init__()メソッドに渡されます。bos_token、eos_token、unk_token、sep_token、pad_token、cls_token、mask_token、additional_special_tokensなどの特殊トークンを設定するために使用できます。詳細については、__init__()のパラメータを参照してください。
事前学習済みモデルの語彙から、ライブラリのトークナイザークラスのいずれかをインスタンス化します。
インスタンス化されるトークナイザーのクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- EvollaModel — TokenizersBackend(EvollaConfigモデル)
- aimv2 — CLIPTokenizer (Aimv2Config モデル)
- albert — AlbertTokenizer(AlbertConfig モデル)
- align — BertTokenizer(AlignConfig モデル)
- aria — TokenizersBackend(AriaConfigモデル)
- audioflamingo3 — Qwen2Tokenizer (AudioFlamingo3Config モデル)
- aya_vision — CohereTokenizer(AyaVisionConfigモデル)
- bark — BertTokenizer(BarkConfig モデル)
- bart — RobertaTokenizer(BartConfigモデル)
- beam — TokenizersBackend(BeamConfig モデル)
- bert — BertTokenizer (BertConfig モデル)
- bert-generation — BertGenerationTokenizer(BertGenerationConfig モデル)
- big_bird — BigBirdTokenizer(BigBirdConfigモデル)
- bigbird_pegasus — TokenizersBackend(BigBirdPegasusConfig モデル)
- biogpt — BioGptTokenizer(BioGptConfig モデル)
- blenderbot — BlenderbotTokenizer(BlenderbotConfig モデル)
- blenderbot-small — BlenderbotSmallTokenizer(BlenderbotSmallConfig モデル)
- blip — BertTokenizer(BlipConfig モデル)
- blip-2 — GPT2Tokenizer(Blip2Config モデル)
- bridgetower — RobertaTokenizer(BridgeTowerConfigモデル)
- bros — BertTokenizer(BrosConfig モデル)
- camembert — CamembertTokenizer(CamembertConfig モデル)
- canary — TokenizersBackend(CanaryConfig モデル)
- canine — CanineTokenizer(CanineConfig モデル)
- chinese_clip — BertTokenizer(ChineseCLIPConfig モデル)
- clap — RobertaTokenizer(ClapConfig モデル)
- clip — CLIPTokenizer(CLIPConfigモデル)
- clipseg — CLIPTokenizer(CLIPSegConfigモデル)
- clvp — ClvpTokenizer(ClvpConfig モデル)
- codegen — GPT2Tokenizer (CodeGenConfig モデル)
- cohere — CohereTokenizer(CohereConfigモデル)
- cohere2 — CohereTokenizer(Cohere2Config モデル)
- cohere_asr — TokenizersBackend(CohereAsrConfig モデル)
- colqwen2 — Qwen2Tokenizer(ColQwen2Config モデル)
- convbert — BertTokenizer(ConvBertConfigモデル)
- cosmos3_omni — Qwen2Tokenizer(Cosmos3OmniConfig モデル)
- cpmant — CpmAntTokenizer(CpmAntConfigモデル)
- ctrl — CTRLTokenizer(CTRLConfig モデル)
- data2vec-audio — Wav2Vec2CTCTokenizer(Data2VecAudioConfig モデル)
- data2vec-text — RobertaTokenizer(Data2VecTextConfigモデル)
- dbrx — GPT2Tokenizer (DbrxConfig モデル)
- deberta — DebertaTokenizer(DebertaConfig モデル)
- deberta-v2 — DebertaV2Tokenizer(DebertaV2Config モデル)
- dia — DiaTokenizer(DiaConfigモデル)
- diffusion_gemma — GemmaTokenizer(DiffusionGemmaConfig モデル)
- distilbert — BertTokenizer(DistilBertConfig モデル)
- dots3_note — Qwen2Tokenizer(Dots3NoteConfigモデル)
- dpr — DPRQuestionEncoderTokenizer (DPRConfig モデル)
- electra — BertTokenizer(ElectraConfigモデル)
- emu3 — GPT2Tokenizer(Emu3Config モデル)
- ernie — BertTokenizer(ErnieConfigモデル)
- ernie4_5 — TokenizersBackend(Ernie4_5Config モデル)
- ernie4_5_moe — TokenizersBackend(Ernie4_5_MoeConfig モデル)
- esm — EsmTokenizer(EsmConfig モデル)
- esmc — EsmcTokenizer(EsmcConfigモデル)
- esmfold2 — EsmcTokenizer(EsmFold2Config モデル)
- falcon_mamba — GPTNeoXTokenizer(FalconMambaConfigモデル)
- fastspeech2_conformer —
None(FastSpeech2ConformerConfig モデル) - flaubert — FlaubertTokenizer(FlaubertConfigモデル)
- flava — BertTokenizer (FlavaConfig モデル)
- flex_olmo — TokenizersBackend(FlexOlmoConfigモデル)
- florence2 — BartTokenizer(Florence2Config モデル)
- fnet — FNetTokenizer(FNetConfig モデル)
- fsmt — FSMTTokenizer(FSMTConfig モデル)
- fun_asr_nano — Qwen2Tokenizer(FunAsrNanoConfig モデル)
- funnel — FunnelTokenizer(FunnelConfigモデル)
- fuyu — TokenizersBackend(FuyuConfigモデル)
- gemma — GemmaTokenizer(GemmaConfig モデル)
- gemma2 — GemmaTokenizer(Gemma2Configモデル)
- gemma3 — GemmaTokenizer(Gemma3Config モデル)
- gemma3_text — GemmaTokenizer(Gemma3TextConfig モデル)
- gemma3n — GemmaTokenizer(Gemma3nConfig モデル)
- gemma3n_text — GemmaTokenizer(Gemma3nTextConfig モデル)
- git — BertTokenizer(GitConfig モデル)
- glm — TokenizersBackend(GlmConfigモデル)
- glm4 — TokenizersBackend(Glm4Config モデル)
- glm4_moe — TokenizersBackend(Glm4MoeConfig モデル)
- glm4_moe_lite — TokenizersBackend(Glm4MoeLiteConfig モデル)
- glm4v — TokenizersBackend(Glm4vConfig モデル)
- glm4v_moe — TokenizersBackend(Glm4vMoeConfig モデル)
- glm_image — TokenizersBackend(GlmImageConfig モデル)
- glmasr — TokenizersBackend(GlmAsrConfig モデル)
- got_ocr2 — TokenizersBackend(GotOcr2Config モデル)
- gpt-sw3 — GPTSw3Tokenizer(GPT2Configモデル)
- gpt2 — GPT2Tokenizer (GPT2Config モデル)
- gpt_bigcode — TokenizersBackend(GPTBigCodeConfigモデル)
- gpt_neo — GPT2Tokenizer(GPTNeoConfig モデル)
- gpt_neox — GPTNeoXTokenizer(GPTNeoXConfigモデル)
- gpt_neox_japanese — GPTNeoXJapaneseTokenizer(GPTNeoXJapaneseConfig モデル)
- gptj — GPT2Tokenizer (GPTJConfig モデル)
- gradium_phonon_tts — TokenizersBackend(GradiumPhononTTSConfig モデル)
- granite — TokenizersBackend(GraniteConfigモデル)
- granite_speech5_ctc — ParakeetTokenizer(GraniteSpeech5CTCConfigモデル)
- granitemoe — TokenizersBackend(GraniteMoeConfig モデル)
- granitemoehybrid — TokenizersBackend(GraniteMoeHybridConfig モデル)
- granitemoeshared — TokenizersBackend(GraniteMoeSharedConfigモデル)
- grounding-dino — BertTokenizer(GroundingDinoConfig モデル)
- groupvit — CLIPTokenizer(GroupViTConfigモデル)
- gte — XLMRobertaTokenizer(GteConfig モデル)
- hubert — Wav2Vec2CTCTokenizer(HubertConfig モデル)
- hunyuan_vl — Qwen2Tokenizer (HunYuanVLConfig モデル)
- hyperclovax — TokenizersBackend(HyperCLOVAXConfigモデル)
- hyperclovax_vision_v2 — GPT2Tokenizer(HyperCLOVAXVisionV2Config モデル)
- ibert — RobertaTokenizer(IBertConfig モデル)
- idefics — LlamaTokenizer(IdeficsConfig モデル)
- idefics2 — LlamaTokenizer(Idefics2Config モデル)
- instructblip — GPT2Tokenizer(InstructBlipConfig モデル)
- instructblipvideo — GPT2Tokenizer(InstructBlipVideoConfig モデル)
- internlm3 — TokenizersBackend(InternLM3Config モデル)
- internvl — Qwen2Tokenizer(InternVLConfig モデル)
- iquestcoder — TokenizersBackend(IQuestCoderConfigモデル)
- jais2 — GPT2Tokenizer(Jais2Config モデル)
- jamba — TokenizersBackend(JambaConfigモデル)
- jina_embeddings_v3 — XLMRobertaTokenizer (JinaEmbeddingsV3Config モデル)
- kimi_k25 — TokenizersBackend(Kimi_K25Config モデル)
- kimi_linear — TokenizersBackend(KimiLinearConfigモデル)
- kosmos-2 — TokenizersBackend(Kosmos2Config モデル)
- lasr_ctc — LasrTokenizer(LasrCTCConfigモデル)
- lasr_encoder — LasrTokenizer(LasrEncoderConfigモデル)
- layoutlm — BertTokenizer(LayoutLMConfig モデル)
- layoutlmv2 — LayoutLMv2Tokenizer(LayoutLMv2Config モデル)
- layoutlmv3 — LayoutLMv3Tokenizer(LayoutLMv3Config モデル)
- layoutxlm — LayoutXLMTokenizer(LayoutXLMConfig モデル)
- led — LEDTokenizer(LEDConfigモデル)
- lighton_ocr — Qwen2TokenizerFast(LightOnOcrConfig モデル)
- lilt — RobertaTokenizer(LiltConfig モデル)
- llava — TokenizersBackend(LlavaConfig モデル)
- llava_next — TokenizersBackend(LlavaNextConfig モデル)
- longformer — RobertaTokenizer(LongformerConfig モデル)
- luke — LukeTokenizer(LukeConfig モデル)
- lxmert — LxmertTokenizer(LxmertConfig モデル)
- m2m_100 — M2M100Tokenizer(M2M100Config モデル)
- mamba — GPTNeoXTokenizer(MambaConfigモデル)
- mamba2 — GPTNeoXTokenizer(Mamba2Config モデル)
- marian — MarianTokenizer(MarianConfigモデル)
- markuplm — MarkupLMTokenizer(MarkupLMConfig モデル)
- mbart — MBartTokenizer(MBartConfig モデル)
- megatron-bert — BertTokenizer(MegatronBertConfig モデル)
- metaclip_2 — XLMRobertaTokenizer(MetaClip2Config モデル)
- mgp-str — MgpstrTokenizer(MgpstrConfig モデル)
- mimo_v2_flash — TokenizersBackend(MiMoV2FlashConfig モデル)
- minicpm3 — TokenizersBackend(MiniCPM3Config モデル)
- minicpm_ocr — TokenizersBackend(MiniCPMOcrConfig モデル)
- minicpmv4_6 — TokenizersBackend(MiniCPMV4_6Config モデル)
- minicpmv4_7 — TokenizersBackend(MiniCPMV4_7Config モデル)
- minimax_m2 — TokenizersBackend(MiniMaxM2Config モデル)
- ministral — MistralCommonBackend(MinistralConfig モデル)
- ministral3 — MistralCommonBackend(Ministral3Config モデル)
- mistral — MistralCommonBackend(MistralConfigモデル)
- mistral3 — MistralCommonBackend(Mistral3Config モデル)
- mistral_large4 — MistralCommonBackend(MistralLarge4Config モデル)
- mixtral — MistralCommonBackend(MixtralConfigモデル)
- mm-grounding-dino — BertTokenizer(MMGroundingDinoConfig モデル)
- mobilebert — MobileBertTokenizer(MobileBertConfig モデル)
- modernbert — TokenizersBackend(ModernBertConfig モデル)
- modernbert-decoder — TokenizersBackend(ModernBertDecoderConfig モデル)
- molmo2 — TokenizersBackend(Molmo2Config モデル)
- moss_transcribe_diarize — Qwen2Tokenizer (MossTranscribeDiarizeConfig モデル)
- mpnet — MPNetTokenizer(MPNetConfigモデル)
- mpt — GPTNeoXTokenizer(MptConfig モデル)
- mra — RobertaTokenizer(MraConfigモデル)
- mt5 — T5Tokenizer(MT5Config モデル)
- musicgen — T5Tokenizer(MusicgenConfig モデル)
- musicgen_melody — T5Tokenizer(MusicgenMelodyConfig モデル)
- mvp — MvpTokenizer(MvpConfig モデル)
- nemo_speechlm — TokenizersBackend(Nemotron3_5TranscribeConfigモデル)
- nemotron — TokenizersBackend(NemotronConfigモデル)
- nemotron3_5_asr — ParakeetTokenizer(Nemotron3_5AsrConfig モデル)
- nemotron_asr_streaming — ParakeetTokenizer(NemotronAsrStreamingConfig モデル)
- nllb-moe — NllbTokenizer(NllbMoeConfig モデル)
- nomic_bert — BertTokenizer(NomicBertConfigモデル)
- nougat — NougatTokenizer(NougatConfigモデル)
- nystromformer — TokenizersBackend(NystromformerConfigモデル)
- olmo — GPTNeoXTokenizer(OlmoConfig モデル)
- olmo2 — TokenizersBackend(Olmo2Config モデル)
- olmo3 — TokenizersBackend(Olmo3Configモデル)
- olmo_hybrid — TokenizersBackend(OlmoHybridConfig モデル)
- olmoe — GPTNeoXTokenizer(OlmoeConfig モデル)
- omdet-turbo — CLIPTokenizer(OmDetTurboConfig モデル)
- oneformer — CLIPTokenizer(OneFormerConfig モデル)
- openai-gpt — OpenAIGPTTokenizer(OpenAIGPTConfigモデル)
- opt — GPT2Tokenizer(OPTConfig モデル)
- ovis2 — Qwen2Tokenizer(Ovis2Config モデル)
- owlv2 — CLIPTokenizer(Owlv2Config モデル)
- owlvit — CLIPTokenizer(OwlViTConfig モデル)
- paddleocr_vl — TokenizersBackend(PaddleOCRVLConfig モデル)
- parakeet_ctc — ParakeetTokenizer(ParakeetCTCConfigモデル)
- parakeet_rnnt — ParakeetTokenizer(ParakeetRNNTConfig モデル)
- parakeet_tdt — ParakeetTokenizer(ParakeetTDTConfig モデル)
- pegasus — PegasusTokenizer(PegasusConfig モデル)
- pegasus_x — TokenizersBackend(PegasusXConfigモデル)
- perceiver — PerceiverTokenizer(PerceiverConfig モデル)
- persimmon — TokenizersBackend(PersimmonConfigモデル)
- phi — GPT2Tokenizer(PhiConfig モデル)
- phi3 — TokenizersBackend(Phi3Config モデル)
- phimoe — TokenizersBackend(PhimoeConfigモデル)
- pix2struct — TokenizersBackend(Pix2StructConfig モデル)
- pixtral — MistralCommonBackend(PixtralVisionConfig モデル)
- plbart — PLBartTokenizer(PLBartConfigモデル)
- pocket_tts — TokenizersBackend(PocketTTSConfig モデル)
- pp_formulanet — NougatTokenizer(PPFormulaNetConfig モデル)
- prophetnet — ProphetNetTokenizer(ProphetNetConfigモデル)
- qianfan_ocr — Qwen2Tokenizer(QianfanOCRConfig モデル)
- qwen2 — Qwen2Tokenizer(Qwen2Config モデル)
- qwen2_5_omni — Qwen2Tokenizer(Qwen2_5OmniConfig モデル)
- qwen2_5_vl — Qwen2Tokenizer(Qwen2_5_VLConfig モデル)
- qwen2_audio — Qwen2Tokenizer(Qwen2AudioConfig モデル)
- qwen2_moe — Qwen2Tokenizer(Qwen2MoeConfig モデル)
- qwen2_vl — Qwen2Tokenizer(Qwen2VLConfig モデル)
- qwen3 — Qwen2Tokenizer(Qwen3Config モデル)
- qwen3_5 — Qwen3_5Tokenizer(Qwen3_5Config モデル)
- qwen3_5_moe — Qwen3_5Tokenizer(Qwen3_5MoeConfig モデル)
- qwen3_5_moe_text — Qwen3_5Tokenizer(Qwen3_5MoeTextConfig モデル)
- qwen3_5_text — Qwen3_5Tokenizer(Qwen3_5TextConfig モデル)
- qwen3_asr — Qwen2Tokenizer(Qwen3ASRConfig モデル)
- qwen3_moe — Qwen2Tokenizer(Qwen3MoeConfig モデル)
- qwen3_next — Qwen2Tokenizer(Qwen3NextConfig モデル)
- qwen3_omni_moe — Qwen2Tokenizer(Qwen3OmniMoeConfig モデル)
- qwen3_vl — Qwen2Tokenizer(Qwen3VLConfig モデル)
- qwen3_vl_moe — Qwen2Tokenizer(Qwen3VLMoeConfig モデル)
- qwen4_exp — Qwen3_5Tokenizer(Qwen4ExpConfig モデル)
- rag — RagTokenizer(RagConfig モデル)
- recurrent_gemma — GemmaTokenizer(RecurrentGemmaConfig モデル)
- reformer — ReformerTokenizer(ReformerConfigモデル)
- rembert — RemBertTokenizer (RemBertConfig モデル)
- roberta — RobertaTokenizer(RobertaConfig モデル)
- roberta-prelayernorm — RobertaTokenizer(RobertaPreLayerNormConfig モデル)
- roc_bert — RoCBertTokenizer(RoCBertConfig モデル)
- roformer — RoFormerTokenizer(RoFormerConfigモデル)
- rwkv — GPTNeoXTokenizer(RwkvConfig モデル)
- sam3 — CLIPTokenizer(Sam3Config モデル)
- sam3_video — CLIPTokenizer (Sam3VideoConfig モデル)
- seamless_m4t — SeamlessM4TTokenizer(SeamlessM4TConfig モデル)
- seamless_m4t_v2 — SeamlessM4TTokenizer(SeamlessM4Tv2Config モデル)
- shieldgemma2 — GemmaTokenizer(ShieldGemma2Config モデル)
- siglip — SiglipTokenizer(SiglipConfig モデル)
- siglip2 — Siglip2Tokenizer(Siglip2Config モデル)
- smolvlm — TokenizersBackend(SmolVLMConfig モデル)
- speech_to_text — Speech2TextTokenizer(Speech2TextConfig モデル)
- speecht5 — SpeechT5Tokenizer(SpeechT5Config モデル)
- splinter — SplinterTokenizer(SplinterConfig モデル)
- squeezebert — BertTokenizer(SqueezeBertConfig モデル)
- stablelm — TokenizersBackend(StableLmConfig モデル)
- starcoder2 — GPT2Tokenizer(Starcoder2Config モデル)
- switch_transformers — T5Tokenizer(SwitchTransformersConfigモデル)
- t5 — T5Tokenizer(T5Config モデル)
- t5gemma — GemmaTokenizer(T5GemmaConfigモデル)
- tapas — TapasTokenizer(TapasConfig モデル)
- tipsv2 — Tipsv2Tokenizer(Tipsv2Configモデル)
- trocr — XLMRobertaTokenizer (TrOCRConfig モデル)
- tvp — BertTokenizer(TvpConfig モデル)
- udop — UdopTokenizer(UdopConfig モデル)
- umt5 — TokenizersBackend(UMT5Config モデル)
- unispeech — Wav2Vec2CTCTokenizer(UniSpeechConfigモデル)
- unispeech-sat — Wav2Vec2CTCTokenizer(UniSpeechSatConfig モデル)
- vibevoice — Qwen2TokenizerFast(VibeVoiceConfig モデル)
- videoprism — VideoPrismTokenizer(VideoPrismConfig モデル)
- vilt — BertTokenizer(ViltConfig モデル)
- vipllava — TokenizersBackend(VipLlavaConfig モデル)
- visual_bert — BertTokenizer(VisualBertConfigモデル)
- vits — VitsTokenizer (VitsConfig モデル)
- voxtral — MistralCommonBackend(VoxtralConfig モデル)
- voxtral_realtime — MistralCommonBackend(VoxtralRealtimeConfig モデル)
- wav2vec2 — Wav2Vec2CTCTokenizer (Wav2Vec2Config モデル)
- wav2vec2-bert — Wav2Vec2CTCTokenizer(Wav2Vec2BertConfig モデル)
- wav2vec2-conformer — Wav2Vec2CTCTokenizer(Wav2Vec2ConformerConfig モデル)
- whisper — WhisperTokenizer(WhisperConfig モデル)
- xclip — CLIPTokenizer(XCLIPConfig モデル)
- xglm — XGLMTokenizer(XGLMConfig モデル)
- xlm — XLMTokenizer(XLMConfig モデル)
- xlm-roberta — XLMRobertaTokenizer(XLMRobertaConfig モデル)
- xlm-roberta-xl — TokenizersBackend(XLMRobertaXLConfig モデル)
- xlnet — XLNetTokenizer (XLNetConfig モデル)
- xlstm — GPTNeoXTokenizer (xLSTMConfig モデル)
- xmod — XLMRobertaTokenizer(XmodConfigモデル)
- yoso — AlbertTokenizer(YosoConfigモデル)
- zaya — GemmaTokenizer(ZayaConfig モデル)
例:
>>> from transformers import AutoTokenizer
>>> # Download vocabulary from huggingface.co and cache.
>>> tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-base-uncased")
>>> # Download vocabulary from huggingface.co (user-uploaded) and cache.
>>> tokenizer = AutoTokenizer.from_pretrained("dbmdz/bert-base-german-cased")
>>> # If vocabulary files are in a directory (e.g. tokenizer was saved using *save_pretrained('./test/saved_model/')*)
>>> # tokenizer = AutoTokenizer.from_pretrained("./test/bert_saved_model/")
>>> # Download vocabulary from huggingface.co and define model-specific arguments
>>> tokenizer = AutoTokenizer.from_pretrained("FacebookAI/roberta-base", add_prefix_space=True)register
< source >( config_classtokenizer_class = Noneslow_tokenizer_class = Nonefast_tokenizer_class = Noneexist_ok = False )
パラメータ
- config_class (PreTrainedConfig) — 登録対象のモデルに対応する設定です。
- tokenizer_class — 登録するトークナイザークラス(V5 - 推奨パラメータ)。
- slow_tokenizer_class — (非推奨) 登録するスロートークナイザ
- fast_tokenizer_class — (非推奨) 登録する fast tokenizer。
このマッピングに新しいトークナイザーを登録します。
AutoFeatureExtractor
これは汎用の特徴量抽出クラスであり、AutoFeatureExtractor.from_pretrained() クラスメソッドで生成された場合、ライブラリの特徴量抽出クラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_pretrained
< source >( pretrained_model_name_or_path**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 次のいずれかです:- 文字列、huggingface.co上のモデルリポジトリにホストされている事前学習済みfeature_extractorのmodel id
- save_pretrained()メソッドを使用して保存されたフィーチャーエクストラクターファイルを含むディレクトリへのパス。例:
./my_model_directory/ - 保存されたフィーチャーエクストラクターの
./my_model_directory/preprocessor_config.jsonのようなファイルへのパス
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルのフィーチャーエクストラクターをキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — feature extractor のファイルを(再)ダウンロードを強制し、キャッシュされたバージョンが存在する場合にそれを上書きするかどうかを指定します。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}.各リクエストでこれらのプロキシが使用されます。 - token (
strまたは bool、省略可) — リモートファイルの HTTP ベアラ認証に使用するトークン。Trueの場合、hf auth loginを実行した際に生成されたトークン(~/.huggingfaceに保存)を使用します。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - return_unused_kwargs (
bool、任意、デフォルトはFalse) —Falseの場合、この関数は最終的な feature extractor オブジェクトのみを返します。Trueの場合、この関数はTuple(feature_extractor, unused_kwargs)を返します。ここで unused_kwargs は、キーが feature extractor の属性でないキー/値のペアで構成される辞書です。つまり、feature_extractorの更新に使用されず、それ以外は無視されるkwargsの部分です。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - kwargs (
dict[str, Any], 任意) — kwargs 内の、feature extractor の属性であるキーの値は、読み込まれた値の上書きに使用されます。キーが feature extractor の属性でないキー/値ペアの動作は、return_unused_kwargsキーワード引数によって制御されます。
事前学習済みモデルの語彙から、ライブラリのフィーチャーエクストラクタークラスのいずれかをインスタンス化します。
インスタンス化されるフィーチャーエクストラクターのクラスは、設定オブジェクトのmodel_typeプロパティに基づいて選択されます(引数として渡された場合、またはpretrained_model_name_or_pathから読み込むことが可能な場合)。model_typeが欠落している場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックします。
- audio-spectrogram-transformer — ASTFeatureExtractor(ASTConfigモデル)
- audioflamingo3 — WhisperFeatureExtractor(AudioFlamingo3Config モデル)
- canary — ParakeetFeatureExtractor(CanaryConfig モデル)
- clap — ClapFeatureExtractor (ClapConfig モデル)
- clvp — ClvpFeatureExtractor(ClvpConfigモデル)
- cohere_asr — CohereAsrFeatureExtractor(CohereAsrConfig モデル)
- csm — EncodecFeatureExtractor(CsmConfigモデル)
- dac — DacFeatureExtractor(DacConfig モデル)
- data2vec-audio — Wav2Vec2FeatureExtractor(Data2VecAudioConfig モデル)
- dia — DiaFeatureExtractor (DiaConfig モデル)
- dots3_note — Dots3NoteFeatureExtractor(Dots3NoteConfigモデル)
- embedding_gemma2 — Gemma4AudioFeatureExtractor(EmbeddingGemma2Config モデル)
- encodec — EncodecFeatureExtractor(EncodecConfig モデル)
- fun_asr_nano — FunAsrNanoFeatureExtractor(FunAsrNanoConfig モデル)
- gemma3n — Gemma3nAudioFeatureExtractor(Gemma3nConfig モデル)
- gemma4 — Gemma4AudioFeatureExtractor(Gemma4Config モデル)
- gemma4_unified — Gemma4UnifiedAudioFeatureExtractor(Gemma4UnifiedConfig モデル)
- glmasr — WhisperFeatureExtractor(GlmAsrConfig モデル)
- gradium_phonon_tts — PocketTTSFeatureExtractor(GradiumPhononTTSConfig モデル)
- granite_speech — GraniteSpeechFeatureExtractor(GraniteSpeechConfig モデル)
- granite_speech5_ctc — GraniteSpeech5FeatureExtractor(GraniteSpeech5CTCConfig モデル)
- granite_speech5_encoder — GraniteSpeech5FeatureExtractor(GraniteSpeech5EncoderConfig モデル)
- granite_speech_plus — GraniteSpeechFeatureExtractor(GraniteSpeechPlusConfig モデル)
- higgs_audio_v2_tokenizer — DacFeatureExtractor(HiggsAudioV2TokenizerConfig モデル)
- hubert — Wav2Vec2FeatureExtractor(HubertConfigモデル)
- inkling_mm_model — InklingFeatureExtractor(InklingConfig モデル)
- kyutai_speech_to_text — KyutaiSpeechToTextFeatureExtractor(KyutaiSpeechToTextConfig モデル)
- lasr_ctc — LasrFeatureExtractor(LasrCTCConfig モデル)
- lasr_encoder — LasrFeatureExtractor(LasrEncoderConfig モデル)
- mimi — EncodecFeatureExtractor(MimiConfigモデル)
- moonshine — Wav2Vec2FeatureExtractor(MoonshineConfig モデル)
- moshi — EncodecFeatureExtractor(MoshiConfig モデル)
- moss_transcribe_diarize — MossTranscribeDiarizeFeatureExtractor(MossTranscribeDiarizeConfig モデル)
- muse_spark — MuseSparkFeatureExtractor(MuseSparkConfigモデル)
- musicgen — EncodecFeatureExtractor (MusicgenConfig モデル)
- musicgen_melody — MusicgenMelodyFeatureExtractor(MusicgenMelodyConfig モデル)
- nemo_speechlm — NemotronAsrStreamingFeatureExtractor(Nemotron3_5TranscribeConfig モデル)
- nemotron3_5_asr — NemotronAsrStreamingFeatureExtractor(Nemotron3_5AsrConfig モデル)
- nemotron3_diarization — NemotronAsrStreamingFeatureExtractor(Nemotron3DiarizationConfig モデル)
- nemotron_asr_streaming — NemotronAsrStreamingFeatureExtractor(NemotronAsrStreamingConfig モデル)
- nemotron_asr_streaming_encoder — NemotronAsrStreamingFeatureExtractor(NemotronAsrStreamingEncoderConfig モデル)
- neucodec — NeuCodecFeatureExtractor(NeuCodecConfigモデル)
- parakeet_ctc — ParakeetFeatureExtractor(ParakeetCTCConfig モデル)
- parakeet_encoder — ParakeetFeatureExtractor(ParakeetEncoderConfig モデル)
- parakeet_rnnt — ParakeetFeatureExtractor(ParakeetRNNTConfigモデル)
- parakeet_tdt — ParakeetFeatureExtractor(ParakeetTDTConfig モデル)
- pe_audio — PeAudioFeatureExtractor(PeAudioConfig モデル)
- pe_audio_video — PeAudioFeatureExtractor(PeAudioVideoConfig モデル)
- phi4_multimodal — Phi4MultimodalFeatureExtractor(Phi4MultimodalConfig モデル)
- pocket_tts — PocketTTSFeatureExtractor(PocketTTSConfig モデル)
- pocket_tts_audio_tokenizer — EncodecFeatureExtractor(PocketTTSAudioTokenizerConfig モデル)
- pop2piano — Pop2PianoFeatureExtractor(Pop2PianoConfig モデル)
- qwen2_5_omni — WhisperFeatureExtractor(Qwen2_5OmniConfig モデル)
- qwen2_audio — WhisperFeatureExtractor(Qwen2AudioConfig モデル)
- qwen3_asr — Qwen3ASRFeatureExtractor(Qwen3ASRConfig モデル)
- qwen3_omni_moe — WhisperFeatureExtractor(Qwen3OmniMoeConfig モデル)
- seamless_m4t — SeamlessM4TFeatureExtractor(SeamlessM4TConfig モデル)
- seamless_m4t_v2 — SeamlessM4TFeatureExtractor(SeamlessM4Tv2Config モデル)
- sew — Wav2Vec2FeatureExtractor(SEWConfig モデル)
- sew-d — Wav2Vec2FeatureExtractor(SEWDConfigモデル)
- speech_to_text — Speech2TextFeatureExtractor(Speech2TextConfig モデル)
- speecht5 — SpeechT5FeatureExtractor(SpeechT5Config モデル)
- unispeech — Wav2Vec2FeatureExtractor(UniSpeechConfig モデル)
- unispeech-sat — Wav2Vec2FeatureExtractor(UniSpeechSatConfig モデル)
- univnet — UnivNetFeatureExtractor(UnivNetConfig モデル)
- vibevoice — VibeVoiceAcousticTokenizerFeatureExtractor(VibeVoiceConfig モデル)
- vibevoice_acoustic_tokenizer — VibeVoiceAcousticTokenizerFeatureExtractor(VibeVoiceAcousticTokenizerConfig モデル)
- vibevoice_asr — VibeVoiceAcousticTokenizerFeatureExtractor(VibeVoiceAsrConfig モデル)
- voxtral — WhisperFeatureExtractor (VoxtralConfig モデル)
- voxtral_realtime — VoxtralRealtimeFeatureExtractor(VoxtralRealtimeConfig モデル)
- wav2vec2 — Wav2Vec2FeatureExtractor(Wav2Vec2Config モデル)
- wav2vec2-bert — Wav2Vec2FeatureExtractor(Wav2Vec2BertConfig モデル)
- wav2vec2-conformer — Wav2Vec2FeatureExtractor(Wav2Vec2ConformerConfig モデル)
- wavlm — Wav2Vec2FeatureExtractor(WavLMConfig モデル)
- whisper — WhisperFeatureExtractor(WhisperConfig モデル)
- xcodec — DacFeatureExtractor(XcodecConfig モデル)
- xcodec2 — Xcodec2FeatureExtractor(Xcodec2Configモデル)
プライベートモデルを使用する場合は、
token=Trueを渡す必要があります。
例:
>>> from transformers import AutoFeatureExtractor
>>> # Download feature extractor from huggingface.co and cache.
>>> feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/wav2vec2-base-960h")
>>> # If feature extractor files are in a directory (e.g. feature extractor was saved using *save_pretrained('./test/saved_model/')*)
>>> # feature_extractor = AutoFeatureExtractor.from_pretrained("./test/saved_model/")register
< source >( config_classfeature_extractor_classexist_ok = False )
パラメータ
- config_class (PreTrainedConfig) — 登録対象のモデルに対応する設定です。
- feature_extractor_class (
FeatureExtractorMixin) — 登録するフィーチャーエクストラクター。
このクラスに新しいフィーチャーエクストラクターを登録します。
AutoImageProcessor
これは汎用の画像プロセッサクラスであり、AutoImageProcessor.from_pretrained() クラスメソッドで生成された場合、ライブラリの画像プロセッサクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_pretrained
< source >( pretrained_model_name_or_path*inputs**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下いずれかです:- 文字列、huggingface.co上のモデルリポジトリにホストされている事前学習済みimage_processorのmodel id。
- a path to a directory containing a image processor file saved using the
save_pretrained() method, e.g.,
./my_model_directory/. - 保存された画像プロセッサの JSON ファイルへのパス。例:
./my_model_directory/preprocessor_config.json
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの画像プロセッサをキャッシュするディレクトリのパス。 - force_download (
bool、省略可、デフォルトはFalse) — イメージプロセッサのファイルを(再)ダウンロードを強制し、キャッシュされたバージョンが存在する場合にそれを上書きするかどうか。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}.各リクエストでこれらのプロキシが使用されます。 - token (
strまたは bool、省略可) — リモートファイルの HTTP ベアラ認証に使用するトークン。Trueの場合、hf auth loginを実行した際に生成されたトークン(~/.huggingfaceに保存)を使用します。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - use_fast (
bool、省略可、デフォルトはFalse) — 非推奨:backend="torchvision"を使用してください。このパラメータは後方互換性のために残されています。 指定されたモデルがサポートしている場合、高速な torchvision ベースの画像プロセッサを使用します。 指定されたモデルに高速な画像プロセッサが利用できない場合、代わりに通常の numpy ベースの画像プロセッサが返されます。 - backend (
str, 省略可, デフォルトはNone) — 画像処理に使用するバックエンド。以下のいずれかを指定できます:None: 利用可能な最適なバックエンドを自動的に選択します(利用可能な場合は torchvision、それ以外は pil)"torchvision": Torchvision バックエンドを使用します(GPU 加速により高速です)"pil": PIL バックエンドを使用(ポータブル、CPU のみ)register()メソッドを介して登録された任意のカスタムバックエンド名
- return_unused_kwargs (
bool、省略可、デフォルトはFalse) —Falseの場合、この関数は最終的な image processor オブジェクトのみを返します。Trueの場合、この関数はTuple(image_processor, unused_kwargs)を返します。ここで unused_kwargs は、キーが image processor の属性でないキー/値のペアで構成される辞書です。つまり、image_processorの更新に使用されず、それ以外は無視されるkwargsの部分です。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - image_processor_filename (
str、任意、デフォルトは"config.json") — モデルディレクトリ内で、画像プロセッサの設定として使用するファイルの名前です。 - kwargs (
dict[str, Any], 省略可) — kwargs 内の、キーが image processor の属性である値は、読み込まれた値の上書きに使用されます。キーが image processor の属性でないキー/値ペアの動作は、return_unused_kwargsキーワード引数によって制御されます。
事前学習済みモデルの語彙から、ライブラリの画像プロセッサクラスのいずれかをインスタンス化します。
インスタンス化される画像プロセッサクラスは、設定オブジェクトのmodel_typeプロパティに基づいて選択されます(引数として渡されるか、可能な場合はpretrained_model_name_or_pathから読み込まれます)。model_typeが欠落している場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックします。
- aimv2 —
{'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'}(Aimv2Config モデル) - aimv2_vision_model —
{'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'}(Aimv2VisionConfig モデル) - align —
{'torchvision': 'EfficientNetImageProcessor', 'pil': 'EfficientNetImageProcessorPil'}(AlignConfig モデル) - altclip —
{'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'}(AltCLIPConfig モデル) - aria —
{'pil': 'AriaImageProcessorPil', 'torchvision': 'AriaImageProcessor'}(AriaConfig モデル) - aya_vision —
{'torchvision': 'GotOcr2ImageProcessor', 'pil': 'GotOcr2ImageProcessorPil'}(AyaVisionConfig モデル) - beit —
{'pil': 'BeitImageProcessorPil', 'torchvision': 'BeitImageProcessor'}(BeitConfig モデル) - bit —
{'pil': 'BitImageProcessorPil', 'torchvision': 'BitImageProcessor'}(BitConfig モデル) - blip —
{'pil': 'BlipImageProcessorPil', 'torchvision': 'BlipImageProcessor'}(BlipConfig モデル) - blip-2 —
{'torchvision': 'BlipImageProcessor', 'pil': 'BlipImageProcessorPil'}(Blip2Config モデル) - bridgetower —
{'pil': 'BridgeTowerImageProcessorPil', 'torchvision': 'BridgeTowerImageProcessor'}(BridgeTowerConfig モデル) - chameleon —
{'pil': 'ChameleonImageProcessorPil', 'torchvision': 'ChameleonImageProcessor'}(ChameleonConfig モデル) - chinese_clip —
{'pil': 'ChineseCLIPImageProcessorPil', 'torchvision': 'ChineseCLIPImageProcessor'}(ChineseCLIPConfig モデル) - chmv2 —
{'torchvision': 'CHMv2ImageProcessor'}(CHMv2Config モデル) - clip —
{'pil': 'CLIPImageProcessorPil', 'torchvision': 'CLIPImageProcessor'}(CLIPConfig モデル) - clipseg —
{'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'}(CLIPSegConfig モデル) - cohere2_vision —
{'torchvision': 'Cohere2VisionImageProcessor'}(Cohere2VisionConfig モデル) - cohere_compass —
{'pil': 'CohereCompassImageProcessorPil', 'torchvision': 'CohereCompassImageProcessor'}(CohereCompassConfig モデル) - colpali —
{'torchvision': 'SiglipImageProcessor', 'pil': 'SiglipImageProcessorPil'}(ColPaliConfig モデル) - colqwen2 —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(ColQwen2Configモデル) - conditional_detr —
{'pil': 'ConditionalDetrImageProcessorPil', 'torchvision': 'ConditionalDetrImageProcessor'}(ConditionalDetrConfig モデル) - convnext —
{'pil': 'ConvNextImageProcessorPil', 'torchvision': 'ConvNextImageProcessor'}(ConvNextConfig モデル) - convnextv2 —
{'torchvision': 'ConvNextImageProcessor', 'pil': 'ConvNextImageProcessorPil'}(ConvNextV2Config モデル) - cosmos3_edge —
{'pil': 'Cosmos3EdgeImageProcessorPil', 'torchvision': 'Cosmos3EdgeImageProcessor'}(Cosmos3EdgeConfig モデル) - cosmos3_omni —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(Cosmos3OmniConfig モデル) - cvt —
{'torchvision': 'ConvNextImageProcessor', 'pil': 'ConvNextImageProcessorPil'}(CvtConfig モデル) - data2vec-vision —
{'torchvision': 'BeitImageProcessor', 'pil': 'BeitImageProcessorPil'}(Data2VecVisionConfig モデル) - deepseek_ocr2 —
{'pil': 'DeepseekOcr2ImageProcessorPil', 'torchvision': 'DeepseekOcr2ImageProcessor'}(DeepseekOcr2Config モデル) - deepseek_vl —
{'pil': 'DeepseekVLImageProcessorPil', 'torchvision': 'DeepseekVLImageProcessor'}(DeepseekVLConfig モデル) - deepseek_vl_hybrid —
{'pil': 'DeepseekVLHybridImageProcessorPil', 'torchvision': 'DeepseekVLHybridImageProcessor'}(DeepseekVLHybridConfig モデル) - deformable_detr —
{'pil': 'DeformableDetrImageProcessorPil', 'torchvision': 'DeformableDetrImageProcessor'}(DeformableDetrConfig モデル) - deimv2 —
{'torchvision': 'RTDetrImageProcessor', 'pil': 'RTDetrImageProcessorPil'}(Deimv2Config モデル) - deit —
{'pil': 'DeiTImageProcessorPil', 'torchvision': 'DeiTImageProcessor'}(DeiTConfig モデル) - depth_anything —
{'torchvision': 'DPTImageProcessor', 'pil': 'DPTImageProcessorPil'}(DepthAnythingConfig モデル) - depth_pro —
{'torchvision': 'DepthProImageProcessor'}(DepthProConfig モデル) - detr —
{'pil': 'DetrImageProcessorPil', 'torchvision': 'DetrImageProcessor'}(DetrConfigモデル) - dinat —
{'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'}(DinatConfig モデル) - dinov2 —
{'torchvision': 'BitImageProcessor', 'pil': 'BitImageProcessorPil'}(Dinov2Config モデル) - dinov3_vit —
{'torchvision': 'DINOv3ViTImageProcessor'}(DINOv3ViTConfig モデル) - donut-swin —
{'torchvision': 'DonutImageProcessor', 'pil': 'DonutImageProcessorPil'}(DonutSwinConfig モデル) - dots3_note —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(Dots3NoteConfig モデル) - dpt —
{'pil': 'DPTImageProcessorPil', 'torchvision': 'DPTImageProcessor'}(DPTConfig モデル) - edgetam —
{'torchvision': 'Sam2ImageProcessor'}(EdgeTamConfig モデル) - efficientloftr —
{'pil': 'EfficientLoFTRImageProcessorPil', 'torchvision': 'EfficientLoFTRImageProcessor'}(EfficientLoFTRConfig モデル) - efficientnet —
{'pil': 'EfficientNetImageProcessorPil', 'torchvision': 'EfficientNetImageProcessor'}(EfficientNetConfig モデル) - embedding_gemma2 —
{'pil': 'Gemma4ImageProcessorPil', 'torchvision': 'Gemma4ImageProcessor'}(EmbeddingGemma2Config モデル) - emu3 —
{'pil': 'Emu3ImageProcessor'}(Emu3Config モデル) - eomt —
{'pil': 'EomtImageProcessorPil', 'torchvision': 'EomtImageProcessor'}(EomtConfig モデル) - eomt_dinov3 —
{'torchvision': 'EomtImageProcessor', 'pil': 'EomtImageProcessorPil'}(EomtDinov3Config モデル) - ernie4_5_vl_moe —
{'pil': 'Ernie4_5_VLMoeImageProcessorPil', 'torchvision': 'Ernie4_5_VLMoeImageProcessor'}(Ernie4_5_VLMoeConfig モデル) - exaone4_5 —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(Exaone4_5_Config モデル) - flava —
{'pil': 'FlavaImageProcessorPil', 'torchvision': 'FlavaImageProcessor'}(FlavaConfig モデル) - florence2 —
{'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'}(Florence2Config モデル) - focalnet —
{'torchvision': 'BitImageProcessor', 'pil': 'BitImageProcessorPil'}(FocalNetConfig モデル) - fuyu —
{'pil': 'FuyuImageProcessorPil', 'torchvision': 'FuyuImageProcessor'}(FuyuConfig モデル) - gemma3 —
{'pil': 'Gemma3ImageProcessorPil', 'torchvision': 'Gemma3ImageProcessor'}(Gemma3Config モデル) - gemma3n —
{'torchvision': 'SiglipImageProcessor', 'pil': 'SiglipImageProcessorPil'}(Gemma3nConfig モデル) - gemma4 —
{'pil': 'Gemma4ImageProcessorPil', 'torchvision': 'Gemma4ImageProcessor'}(Gemma4Config モデル) - gemma4_unified —
{'torchvision': 'Gemma4UnifiedImageProcessor'}(Gemma4UnifiedConfig モデル) - git —
{'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'}(GitConfig モデル) - glm46v —
{'pil': 'Glm46VImageProcessorPil', 'torchvision': 'Glm46VImageProcessor'}(Glm46VConfig モデル) - glm4v —
{'pil': 'Glm4vImageProcessorPil', 'torchvision': 'Glm4vImageProcessor'}(Glm4vConfig モデル) - glm5_next —
{'pil': 'Glm5NextImageProcessorPil', 'torchvision': 'Glm5NextImageProcessor'}(Glm5NextConfig モデル) - glm_image —
{'pil': 'GlmImageImageProcessorPil', 'torchvision': 'GlmImageImageProcessor'}(GlmImageConfig モデル) - glmga —
{'pil': 'GlmgaImageProcessorPil', 'torchvision': 'GlmgaImageProcessor'}(GlmgaConfig モデル) - glpn —
{'pil': 'GLPNImageProcessorPil', 'torchvision': 'GLPNImageProcessor'}(GLPNConfig モデル) - got_ocr2 —
{'pil': 'GotOcr2ImageProcessorPil', 'torchvision': 'GotOcr2ImageProcessor'}(GotOcr2Config モデル) - granite4_vision —
{'torchvision': 'LlavaNextImageProcessor', 'pil': 'LlavaNextImageProcessorPil'}(Granite4VisionConfig モデル) - granite_docling —
{'pil': 'GraniteDoclingImageProcessorPil', 'torchvision': 'GraniteDoclingImageProcessor'}(GraniteDoclingConfig モデル) - grounding-dino —
{'pil': 'GroundingDinoImageProcessorPil', 'torchvision': 'GroundingDinoImageProcessor'}(GroundingDinoConfig モデル) - groupvit —
{'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'}(GroupViTConfig モデル) - hiera —
{'torchvision': 'BitImageProcessor', 'pil': 'BitImageProcessorPil'}(HieraConfig モデル) - hunyuan_vl —
{'pil': 'HunYuanVLImageProcessorPil', 'torchvision': 'HunYuanVLImageProcessor'}(HunYuanVLConfig モデル) - hyperclovax_vision_v2 —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(HyperCLOVAXVisionV2Config モデル) - idefics —
{'pil': 'IdeficsImageProcessorPil', 'torchvision': 'IdeficsImageProcessor'}(IdeficsConfig モデル) - idefics2 —
{'pil': 'Idefics2ImageProcessorPil', 'torchvision': 'Idefics2ImageProcessor'}(Idefics2Config モデル) - idefics3 —
{'pil': 'Idefics3ImageProcessorPil', 'torchvision': 'Idefics3ImageProcessor'}(Idefics3Config モデル) - ijepa —
{'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'}(IJepaConfig モデル) - imagegpt —
{'pil': 'ImageGPTImageProcessorPil', 'torchvision': 'ImageGPTImageProcessor'}(ImageGPTConfig モデル) - inkling2 —
{'torchvision': 'Inkling2ImageProcessor'}(Inkling2Config モデル) - inkling_mm_model —
{'torchvision': 'InklingImageProcessor'}(InklingConfig モデル) - instructblip —
{'torchvision': 'BlipImageProcessor', 'pil': 'BlipImageProcessorPil'}(InstructBlipConfig モデル) - internvl —
{'torchvision': 'GotOcr2ImageProcessor', 'pil': 'GotOcr2ImageProcessorPil'}(InternVLConfig モデル) - janus —
{'pil': 'JanusImageProcessorPil', 'torchvision': 'JanusImageProcessor'}(JanusConfig モデル) - kimi_k25 —
{'torchvision': 'Kimi_K25ImageProcessor'}(Kimi_K25Config モデル) - kosmos-2 —
{'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'}(Kosmos2Config モデル) - kosmos-2.5 —
{'torchvision': 'Kosmos2_5ImageProcessor', 'pil': 'Kosmos2_5ImageProcessorPil'}(Kosmos2_5Config モデル) - layoutlmv2 —
{'pil': 'LayoutLMv2ImageProcessorPil', 'torchvision': 'LayoutLMv2ImageProcessor'}(LayoutLMv2Config モデル) - layoutlmv3 —
{'pil': 'LayoutLMv3ImageProcessorPil', 'torchvision': 'LayoutLMv3ImageProcessor'}(LayoutLMv3Config モデル) - layoutxlm —
{'torchvision': 'LayoutLMv2ImageProcessor', 'pil': 'LayoutLMv2ImageProcessorPil'}(LayoutXLMConfig モデル) - levit —
{'pil': 'LevitImageProcessorPil', 'torchvision': 'LevitImageProcessor'}(LevitConfig モデル) - lfm2_vl —
{'torchvision': 'Lfm2VlImageProcessor'}(Lfm2VlConfig モデル) - lfm3_vl —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(Lfm3VlConfig モデル) - lightglue —
{'pil': 'LightGlueImageProcessorPil', 'torchvision': 'LightGlueImageProcessor'}(LightGlueConfig モデル) - lighton_ocr —
{'torchvision': 'PixtralImageProcessor', 'pil': 'PixtralImageProcessorPil'}(LightOnOcrConfigモデル) - llama4 —
{'torchvision': 'Llama4ImageProcessor'}(Llama4Config モデル) - llava —
{'pil': 'LlavaImageProcessorPil', 'torchvision': 'LlavaImageProcessor'}(LlavaConfigモデル) - llava_next —
{'pil': 'LlavaNextImageProcessorPil', 'torchvision': 'LlavaNextImageProcessor'}(LlavaNextConfig モデル) - llava_next_video —
{'torchvision': 'LlavaNextImageProcessor', 'pil': 'LlavaNextImageProcessorPil'}(LlavaNextVideoConfig モデル) - llava_onevision —
{'pil': 'LlavaOnevisionImageProcessorPil', 'torchvision': 'LlavaOnevisionImageProcessor'}(LlavaOnevisionConfig モデル) - lw_detr —
{'torchvision': 'DeformableDetrImageProcessor', 'pil': 'DeformableDetrImageProcessorPil'}(LwDetrConfig モデル) - mask2former —
{'pil': 'Mask2FormerImageProcessorPil', 'torchvision': 'Mask2FormerImageProcessor'}(Mask2FormerConfig モデル) - maskformer —
{'pil': 'MaskFormerImageProcessorPil', 'torchvision': 'MaskFormerImageProcessor'}(MaskFormerConfigモデル) - metaclip_2 —
{'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'}(MetaClip2Config モデル) - mgp-str —
{'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'}(MgpstrConfig モデル) - minicpm_ocr —
{'torchvision': 'MiniCPMOcrImageProcessor'}(MiniCPMOcrConfig モデル) - minicpmv4_6 —
{'pil': 'MiniCPMV4_6ImageProcessorPil', 'torchvision': 'MiniCPMV4_6ImageProcessor'}(MiniCPMV4_6Config モデル) - minicpmv4_7 —
{'pil': 'MiniCPMV4_6ImageProcessorPil', 'torchvision': 'MiniCPMV4_6ImageProcessor'}(MiniCPMV4_7Config モデル) - minimax_m3_vl —
{'torchvision': 'MiniMaxM3VLImageProcessor'}(MiniMaxM3VLConfig モデル) - mistral3 —
{'torchvision': 'PixtralImageProcessor', 'pil': 'PixtralImageProcessorPil'}(Mistral3Config モデル) - mistral_large4 —
{'torchvision': 'PixtralImageProcessor', 'pil': 'PixtralImageProcessorPil'}(MistralLarge4Config モデル) - mlcd —
{'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'}(MLCDVisionConfig モデル) - mllama —
{'pil': 'MllamaImageProcessorPil', 'torchvision': 'MllamaImageProcessor'}(MllamaConfig モデル) - mm-grounding-dino —
{'torchvision': 'GroundingDinoImageProcessor', 'pil': 'GroundingDinoImageProcessorPil'}(MMGroundingDinoConfig モデル) - mobilenet_v1 —
{'pil': 'MobileNetV1ImageProcessorPil', 'torchvision': 'MobileNetV1ImageProcessor'}(MobileNetV1Config モデル) - mobilenet_v2 —
{'pil': 'MobileNetV2ImageProcessorPil', 'torchvision': 'MobileNetV2ImageProcessor'}(MobileNetV2Config モデル) - mobilevit —
{'pil': 'MobileViTImageProcessorPil', 'torchvision': 'MobileViTImageProcessor'}(MobileViTConfig モデル) - mobilevitv2 —
{'torchvision': 'MobileViTImageProcessor', 'pil': 'MobileViTImageProcessorPil'}(MobileViTV2Config モデル) - molmo2 —
{'torchvision': 'Molmo2ImageProcessor'}(Molmo2Config モデル) - muse_glimmer —
{'torchvision': 'MuseGlimmerImageProcessor'}(MuseGlimmerConfig モデル) - muse_spark —
{'torchvision': 'MuseGlimmerImageProcessor'}(MuseSparkConfig モデル) - nemotron_h_omni —
{'torchvision': 'NemotronH_Omni_Reasoning_V3ImageProcessor'}(NemotronH_Omni_Reasoning_V3_Config モデル) - neomme —
{'torchvision': 'NeoMMEImageProcessor'}(NeoMMEConfig モデル) - nougat —
{'pil': 'NougatImageProcessorPil', 'torchvision': 'NougatImageProcessor'}(NougatConfigモデル) - omdet-turbo —
{'torchvision': 'DetrImageProcessor', 'pil': 'DetrImageProcessorPil'}(OmDetTurboConfig モデル) - oneformer —
{'pil': 'OneFormerImageProcessorPil', 'torchvision': 'OneFormerImageProcessor'}(OneFormerConfig モデル) - ovis2 —
{'pil': 'Ovis2ImageProcessorPil', 'torchvision': 'Ovis2ImageProcessor'}(Ovis2Config モデル) - owlv2 —
{'pil': 'Owlv2ImageProcessorPil', 'torchvision': 'Owlv2ImageProcessor'}(Owlv2Config モデル) - owlvit —
{'pil': 'OwlViTImageProcessorPil', 'torchvision': 'OwlViTImageProcessor'}(OwlViTConfig モデル) - paddleocr_vl —
{'pil': 'PaddleOCRVLImageProcessorPil', 'torchvision': 'PaddleOCRVLImageProcessor'}(PaddleOCRVLConfig モデル) - paligemma —
{'torchvision': 'SiglipImageProcessor', 'pil': 'SiglipImageProcessorPil'}(PaliGemmaConfig モデル) - perceiver —
{'pil': 'PerceiverImageProcessorPil', 'torchvision': 'PerceiverImageProcessor'}(PerceiverConfig モデル) - perception_lm —
{'torchvision': 'PerceptionLMImageProcessor'}(PerceptionLMConfig モデル) - phi4_multimodal —
{'torchvision': 'Phi4MultimodalImageProcessor'}(Phi4MultimodalConfig モデル) - pi0 —
{'torchvision': 'PI0ImageProcessor'}(PI0Config モデル) - pix2struct —
{'pil': 'Pix2StructImageProcessorPil', 'torchvision': 'Pix2StructImageProcessor'}(Pix2StructConfig モデル) - pixio —
{'torchvision': 'BitImageProcessor', 'pil': 'BitImageProcessorPil'}(PixioConfig モデル) - pixtral —
{'pil': 'PixtralImageProcessorPil', 'torchvision': 'PixtralImageProcessor'}(PixtralVisionConfig モデル) - poolformer —
{'pil': 'PoolFormerImageProcessorPil', 'torchvision': 'PoolFormerImageProcessor'}(PoolFormerConfig モデル) - pp_chart2table —
{'pil': 'PPChart2TableImageProcessorPil', 'torchvision': 'PPChart2TableImageProcessor'}(PPChart2TableConfig モデル) - pp_doclayout_v2 —
{'torchvision': 'PPDocLayoutV2ImageProcessor'}(PPDocLayoutV2Config モデル) - pp_doclayout_v3 —
{'torchvision': 'PPDocLayoutV3ImageProcessor'}(PPDocLayoutV3Config モデル) - pp_doclayout_v4 —
{'torchvision': 'PPDocLayoutV4ImageProcessor'}(PPDocLayoutV4Config モデル) - pp_formulanet —
{'torchvision': 'PPFormulaNetImageProcessor'}(PPFormulaNetConfig モデル) - pp_lcnet —
{'torchvision': 'PPLCNetImageProcessor'}(PPLCNetConfig モデル) - pp_ocrv5_mobile_det —
{'torchvision': 'PPOCRV5ServerDetImageProcessor'}(PPOCRV5MobileDetConfig モデル) - pp_ocrv5_mobile_rec —
{'torchvision': 'PPOCRV5ServerRecImageProcessor'}(PPOCRV5MobileRecConfig モデル) - pp_ocrv5_server_det —
{'torchvision': 'PPOCRV5ServerDetImageProcessor'}(PPOCRV5ServerDetConfig モデル) - pp_ocrv5_server_rec —
{'torchvision': 'PPOCRV5ServerRecImageProcessor'}(PPOCRV5ServerRecConfig モデル) - pp_ocrv6_medium_det —
{'torchvision': 'PPOCRV5ServerDetImageProcessor'}(PPOCRV6MediumDetConfig モデル) - pp_ocrv6_small_det —
{'torchvision': 'PPOCRV5ServerDetImageProcessor'}(PPOCRV6SmallDetConfig モデル) - pp_ocrv6_small_rec —
{'torchvision': 'PPOCRV6SmallRecImageProcessor'}(PPOCRV6SmallRecConfig モデル) - pp_ocrv6_tiny_rec —
{'torchvision': 'PPOCRV6SmallRecImageProcessor'}(PPOCRV6TinyRecConfig モデル) - prompt_depth_anything —
{'pil': 'PromptDepthAnythingImageProcessorPil', 'torchvision': 'PromptDepthAnythingImageProcessor'}(PromptDepthAnythingConfig モデル) - pvt —
{'pil': 'PvtImageProcessorPil', 'torchvision': 'PvtImageProcessor'}(PvtConfig モデル) - pvt_v2 —
{'torchvision': 'PvtImageProcessor', 'pil': 'PvtImageProcessorPil'}(PvtV2Config モデル) - qianfan_ocr —
{'torchvision': 'GotOcr2ImageProcessor', 'pil': 'GotOcr2ImageProcessorPil'}(QianfanOCRConfig モデル) - qwen2_5_omni —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(Qwen2_5OmniConfig モデル) - qwen2_5_vl —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(Qwen2_5_VLConfig モデル) - qwen2_vl —
{'pil': 'Qwen2VLImageProcessorPil', 'torchvision': 'Qwen2VLImageProcessor'}(Qwen2VLConfig モデル) - qwen3_5 —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(Qwen3_5Config モデル) - qwen3_5_moe —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(Qwen3_5MoeConfig モデル) - qwen3_omni_moe —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(Qwen3OmniMoeConfig モデル) - qwen3_vl —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(Qwen3VLConfig モデル) - qwen4_exp —
{'torchvision': 'Qwen2VLImageProcessor', 'pil': 'Qwen2VLImageProcessorPil'}(Qwen4ExpConfig モデル) - regnet —
{'torchvision': 'ConvNextImageProcessor', 'pil': 'ConvNextImageProcessorPil'}(RegNetConfigモデル) - resnet —
{'torchvision': 'ConvNextImageProcessor', 'pil': 'ConvNextImageProcessorPil'}(ResNetConfig モデル) - rf_detr —
{'torchvision': 'RfDetrImageProcessor'}(RfDetrConfig モデル) - rt_detr —
{'pil': 'RTDetrImageProcessorPil', 'torchvision': 'RTDetrImageProcessor'}(RTDetrConfig モデル) - sam —
{'pil': 'SamImageProcessorPil', 'torchvision': 'SamImageProcessor'}(SamConfig モデル) - sam2 —
{'torchvision': 'Sam2ImageProcessor'}(Sam2Config モデル) - sam2_video —
{'torchvision': 'Sam2ImageProcessor'}(Sam2VideoConfig モデル) - sam3 —
{'torchvision': 'Sam3ImageProcessor'}(Sam3Config モデル) - sam3_lite_text —
{'torchvision': 'Sam3ImageProcessor'}(Sam3LiteTextConfig モデル) - sam3_tracker —
{'torchvision': 'Sam3ImageProcessor'}(Sam3TrackerConfig モデル) - sam3_tracker_video —
{'torchvision': 'Sam3ImageProcessor'}(Sam3TrackerVideoConfig モデル) - sam3_video —
{'torchvision': 'Sam3ImageProcessor'}(Sam3VideoConfig モデル) - sam_hq —
{'torchvision': 'SamImageProcessor', 'pil': 'SamImageProcessorPil'}(SamHQConfig モデル) - sapiens2 —
{'torchvision': 'Sapiens2ImageProcessor'}(Sapiens2Config モデル) - segformer —
{'pil': 'SegformerImageProcessorPil', 'torchvision': 'SegformerImageProcessor'}(SegformerConfig モデル) - seggpt —
{'pil': 'SegGptImageProcessorPil', 'torchvision': 'SegGptImageProcessor'}(SegGptConfig モデル) - shieldgemma2 —
{'torchvision': 'Gemma3ImageProcessor', 'pil': 'Gemma3ImageProcessorPil'}(ShieldGemma2Config モデル) - siglip —
{'pil': 'SiglipImageProcessorPil', 'torchvision': 'SiglipImageProcessor'}(SiglipConfig モデル) - siglip2 —
{'pil': 'Siglip2ImageProcessorPil', 'torchvision': 'Siglip2ImageProcessor'}(Siglip2Config モデル) - slanet —
{'torchvision': 'SLANeXtImageProcessor'}(SLANetConfigモデル) - slanext —
{'torchvision': 'SLANeXtImageProcessor'}(SLANeXtConfig モデル) - smolvlm —
{'pil': 'SmolVLMImageProcessorPil', 'torchvision': 'SmolVLMImageProcessor'}(SmolVLMConfig モデル) - step3p7 —
{'torchvision': 'Step3p7ImageProcessor'}(Step3p7Config モデル) - superglue —
{'pil': 'SuperGlueImageProcessorPil', 'torchvision': 'SuperGlueImageProcessor'}(SuperGlueConfig モデル) - superpoint —
{'pil': 'SuperPointImageProcessorPil', 'torchvision': 'SuperPointImageProcessor'}(SuperPointConfig モデル) - swiftformer —
{'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'}(SwiftFormerConfig モデル) - swin —
{'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'}(SwinConfig モデル) - swin2sr —
{'pil': 'Swin2SRImageProcessorPil', 'torchvision': 'Swin2SRImageProcessor'}(Swin2SRConfig モデル) - swinv2 —
{'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'}(Swinv2Config モデル) - t5gemma2 —
{'torchvision': 'Gemma3ImageProcessor', 'pil': 'Gemma3ImageProcessorPil'}(T5Gemma2Config モデル) - t5gemma2_encoder —
{'torchvision': 'Gemma3ImageProcessor', 'pil': 'Gemma3ImageProcessorPil'}(T5Gemma2EncoderConfig モデル) - table-transformer —
{'torchvision': 'DetrImageProcessor', 'pil': 'DetrImageProcessorPil'}(TableTransformerConfig モデル) - textnet —
{'pil': 'TextNetImageProcessorPil', 'torchvision': 'TextNetImageProcessor'}(TextNetConfig モデル) - timesformer —
{'pil': 'VideoMAEImageProcessorPil', 'torchvision': 'VideoMAEImageProcessor'}(TimesformerConfig モデル) - timm_wrapper —
{'pil': 'TimmWrapperImageProcessor'}(TimmWrapperConfig モデル) - tipsv2 —
{'torchvision': 'Tipsv2ImageProcessor'}(Tipsv2Config モデル) - tipsv2_dpt —
{'torchvision': 'Tipsv2DptImageProcessor'}(Tipsv2DptConfig モデル) - trocr —
{'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'}(TrOCRConfig モデル) - tvp —
{'pil': 'TvpImageProcessorPil', 'torchvision': 'TvpImageProcessor'}(TvpConfig モデル) - udop —
{'torchvision': 'LayoutLMv3ImageProcessor', 'pil': 'LayoutLMv3ImageProcessorPil'}(UdopConfig モデル) - ultravbert —
{'torchvision': 'UltraVBertImageProcessor'}(UltraVBertConfig モデル) - unlimited_ocr —
{'torchvision': 'UnlimitedOcrImageProcessor'}(UnlimitedOcrConfig モデル) - upernet —
{'torchvision': 'SegformerImageProcessor', 'pil': 'SegformerImageProcessorPil'}(UperNetConfig モデル) - uvdoc —
{'torchvision': 'UVDocImageProcessor'}(UVDocConfig モデル) - video_llama_3 —
{'pil': 'VideoLlama3ImageProcessorPil', 'torchvision': 'VideoLlama3ImageProcessor'}(VideoLlama3Config モデル) - video_llava —
{'pil': 'VideoLlavaImageProcessor'}(VideoLlavaConfig モデル) - videomae —
{'pil': 'VideoMAEImageProcessorPil', 'torchvision': 'VideoMAEImageProcessor'}(VideoMAEConfig モデル) - vilt —
{'pil': 'ViltImageProcessorPil', 'torchvision': 'ViltImageProcessor'}(ViltConfig モデル) - vipllava —
{'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'}(VipLlavaConfig モデル) - vit —
{'pil': 'ViTImageProcessorPil', 'torchvision': 'ViTImageProcessor'}(ViTConfig モデル) - vit_mae —
{'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'}(ViTMAEConfig モデル) - vit_msn —
{'torchvision': 'ViTImageProcessor', 'pil': 'ViTImageProcessorPil'}(ViTMSNConfig モデル) - vitmatte —
{'pil': 'VitMatteImageProcessorPil', 'torchvision': 'VitMatteImageProcessor'}(VitMatteConfig モデル) - vitpose —
{'pil': 'VitPoseImageProcessorPil', 'torchvision': 'VitPoseImageProcessor'}(VitPoseConfig モデル) - vivit —
{'torchvision': 'VivitImageProcessor'}(VivitConfig モデル) - xclip —
{'torchvision': 'CLIPImageProcessor', 'pil': 'CLIPImageProcessorPil'}(XCLIPConfig モデル) - yolos —
{'pil': 'YolosImageProcessorPil', 'torchvision': 'YolosImageProcessor'}(YolosConfig モデル) - zoedepth —
{'pil': 'ZoeDepthImageProcessorPil', 'torchvision': 'ZoeDepthImageProcessor'}(ZoeDepthConfig モデル)
プライベートモデルを使用する場合は、
token=Trueを渡す必要があります。
例:
>>> from transformers import AutoImageProcessor
>>> # Download image processor from huggingface.co and cache.
>>> image_processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k")
>>> # If image processor files are in a directory (e.g. image processor was saved using *save_pretrained('./test/saved_model/')*)
>>> # image_processor = AutoImageProcessor.from_pretrained("./test/saved_model/")register
< source >( config_classslow_image_processor_class: type | None = Nonefast_image_processor_class: type | None = Noneimage_processor_classes: dict[str, type] | None = Noneexist_ok: bool = False )
パラメータ
- config_class (PreTrainedConfig) — 登録対象のモデルに対応する設定です。
- slow_image_processor_class (
type、省略可) — PIL バックエンドの画像プロセッサクラスです(非推奨。image_processor_classes={"pil": ...}を使用してください)。 - fast_image_processor_class (
type、省略可) — Torchvision バックエンドの画像プロセッサクラスです(非推奨。image_processor_classes={"torchvision": ...}を使用してください)。 - image_processor_classes (
dict[str, type]、省略可) — バックエンド名をキーに、画像プロセッサクラスを値とする辞書。カスタムバックエンドの登録を許可します。 例:{"pil": MyPilProcessor, "torchvision": MyTorchvisionProcessor, "custom": MyCustomProcessor} - exist_ok (
bool、省略可、デフォルトはFalse) —Trueの場合、既存の登録を上書きすることを許可します。
このクラスに新しい画像プロセッサを登録します。
AutoVideoProcessor
バックエンドの選択、カスタム登録、および保存と再読み込みについては、Video processor backendsを参照してください。
これは汎用のビデオプロセッサクラスであり、AutoVideoProcessor.from_pretrained() クラスメソッドで生成されると、ライブラリのビデオプロセッサクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_pretrained
< source >( pretrained_model_name_or_path*inputs**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 次のいずれかです:- 文字列、huggingface.co上のモデルリポジトリにホストされている事前学習済みvideo_processorのmodel id
- save_pretrained() メソッドを使用して保存されたビデオプロセッサファイルを含むディレクトリへのパス。例:
./my_model_directory/ - 保存されたビデオプロセッサの JSON ファイルへのパス。例:
./my_model_directory/preprocessor_config.json
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの video processor をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — 存在する場合はキャッシュされたバージョンを上書きし、動画プロセッサのファイルを(再)ダウンロードすることを強制するかどうか。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}.各リクエストでこれらのプロキシが使用されます。 - token (
strまたは bool、省略可) — リモートファイルの HTTP ベアラ認証に使用するトークン。Trueの場合、hf auth loginを実行した際に生成されたトークン(~/.huggingfaceに保存)を使用します。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - return_unused_kwargs (
bool、省略可、デフォルトはFalse) —Falseの場合、この関数は最終的な video processor オブジェクトのみを返します。Trueの場合、この関数はTuple(video_processor, unused_kwargs)を返します。ここで unused_kwargs は、キーが video processor の属性でないキー/値のペアで構成される辞書です。つまり、video_processorの更新に使用されず、それ以外は無視されるkwargsの部分です。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - kwargs (
dict[str, Any]、省略可) — kwargs 内の、キーが video processor の属性である値は、読み込まれた値の上書きに使用されます。キーが video processor の属性でないキー/値ペアの動作は、return_unused_kwargsキーワード引数によって制御されます。
事前学習済みモデルの語彙から、ライブラリの動画プロセッサクラスのいずれかをインスタンス化します。
インスタンス化されるビデオプロセッサのクラスは、設定オブジェクトのmodel_typeプロパティに基づいて選択されます(引数として渡されるか、可能な場合はpretrained_model_name_or_pathから読み込まれます)。model_typeが欠落している場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックします。
- cohere_compass —
{'torchvision': 'CohereCompassVideoProcessor'}(CohereCompassConfig モデル) - cosmos3_edge —
{'torchvision': 'Cosmos3EdgeVideoProcessor'}(Cosmos3EdgeConfig モデル) - cosmos3_omni —
{'torchvision': 'Qwen3VLVideoProcessor'}(Cosmos3OmniConfig モデル) - dots3_note —
{'torchvision': 'Qwen3VLVideoProcessor'}(Dots3NoteConfigモデル) - embedding_gemma2 —
{'torchvision': 'EmbeddingGemma2VideoProcessor'}(EmbeddingGemma2Config モデル) - ernie4_5_vl_moe —
{'torchvision': 'Ernie4_5_VLMoeVideoProcessor'}(Ernie4_5_VLMoeConfig モデル) - exaone4_5 —
{'torchvision': 'Qwen2VLVideoProcessor'}(Exaone4_5_Config モデル) - gemma4 —
{'torchvision': 'Gemma4VideoProcessor'}(Gemma4Config モデル) - gemma4_unified —
{'torchvision': 'Gemma4UnifiedVideoProcessor'}(Gemma4UnifiedConfig モデル) - glm46v —
{'torchvision': 'Glm46VVideoProcessor'}(Glm46VConfig モデル) - glm4v —
{'torchvision': 'Glm4vVideoProcessor'}(Glm4vConfig モデル) - glm5_next —
{'torchvision': 'Glm5NextVideoProcessor'}(Glm5NextConfig モデル) - glmga —
{'torchvision': 'GlmgaVideoProcessor'}(GlmgaConfig モデル) - hyperclovax_vision_v2 —
{'torchvision': 'Qwen2VLVideoProcessor'}(HyperCLOVAXVisionV2Config モデル) - instructblip —
{'torchvision': 'InstructBlipVideoVideoProcessor'}(InstructBlipConfig モデル) - instructblipvideo —
{'torchvision': 'InstructBlipVideoVideoProcessor'}(InstructBlipVideoConfig モデル) - internvl —
{'torchvision': 'InternVLVideoProcessor'}(InternVLConfig モデル) - kimi_k25 —
{'torchvision': 'Kimi_K25VideoProcessor'}(Kimi_K25Config モデル) - lfm3_vl —
{'torchvision': 'Qwen3VLVideoProcessor'}(Lfm3VlConfig モデル) - llava_next_video —
{'torchvision': 'LlavaNextVideoVideoProcessor'}(LlavaNextVideoConfig モデル) - llava_onevision —
{'torchvision': 'LlavaOnevisionVideoProcessor'}(LlavaOnevisionConfig モデル) - minicpm_ocr —
{'torchvision': 'MiniCPMOcrVideoProcessor'}(MiniCPMOcrConfig モデル) - minicpmv4_6 —
{'torchvision': 'MiniCPMV4_6VideoProcessor'}(MiniCPMV4_6Config モデル) - minicpmv4_7 —
{'torchvision': 'MiniCPMV4_6VideoProcessor'}(MiniCPMV4_7Config モデル) - minimax_m3_vl —
{'torchvision': 'MiniMaxM3VLVideoProcessor'}(MiniMaxM3VLConfig モデル) - molmo2 —
{'torchvision': 'Molmo2VideoProcessor'}(Molmo2Config モデル) - muse_glimmer —
{'torchvision': 'MuseGlimmerVideoProcessor'}(MuseGlimmerConfig モデル) - muse_spark —
{'torchvision': 'MuseGlimmerVideoProcessor'}(MuseSparkConfig モデル) - nemotron_h_omni —
{'torchvision': 'NemotronH_Omni_Reasoning_V3VideoProcessor'}(NemotronH_Omni_Reasoning_V3_Config モデル) - pe_audio_video —
{'torchvision': 'PeVideoVideoProcessor'}(PeAudioVideoConfig モデル) - pe_video —
{'torchvision': 'PeVideoVideoProcessor'}(PeVideoConfig モデル) - perception_lm —
{'torchvision': 'PerceptionLMVideoProcessor'}(PerceptionLMConfig モデル) - qwen2_5_omni —
{'torchvision': 'Qwen2VLVideoProcessor'}(Qwen2_5OmniConfig モデル) - qwen2_5_vl —
{'torchvision': 'Qwen2VLVideoProcessor'}(Qwen2_5_VLConfig モデル) - qwen2_vl —
{'torchvision': 'Qwen2VLVideoProcessor'}(Qwen2VLConfig モデル) - qwen3_5 —
{'torchvision': 'Qwen3VLVideoProcessor'}(Qwen3_5Config モデル) - qwen3_5_moe —
{'torchvision': 'Qwen3VLVideoProcessor'}(Qwen3_5MoeConfig モデル) - qwen3_omni_moe —
{'torchvision': 'Qwen2VLVideoProcessor'}(Qwen3OmniMoeConfig モデル) - qwen3_vl —
{'torchvision': 'Qwen3VLVideoProcessor'}(Qwen3VLConfig モデル) - qwen3_vl_moe —
{'torchvision': 'Qwen3VLVideoProcessor'}(Qwen3VLMoeConfig モデル) - qwen4_exp —
{'torchvision': 'Qwen3VLVideoProcessor'}(Qwen4ExpConfig モデル) - sam2_video —
{'torchvision': 'Sam2VideoVideoProcessor'}(Sam2VideoConfig モデル) - smolvlm —
{'torchvision': 'SmolVLMVideoProcessor'}(SmolVLMConfig モデル) - timesformer —
{'torchvision': 'TimesformerVideoProcessor'}(TimesformerConfig モデル) - video_llama_3 —
{'torchvision': 'VideoLlama3VideoProcessor'}(VideoLlama3Config モデル) - video_llava —
{'torchvision': 'VideoLlavaVideoProcessor'}(VideoLlavaConfig モデル) - videomae —
{'torchvision': 'VideoMAEVideoProcessor'}(VideoMAEConfig モデル) - videomt —
{'torchvision': 'VideomtVideoProcessor'}(VideomtConfig モデル) - videoprism —
{'torchvision': 'LlavaOnevisionVideoProcessor'}(VideoPrismConfig モデル) - vivit —
{'torchvision': 'VivitVideoProcessor'}(VivitConfig モデル) - vjepa2 —
{'torchvision': 'VJEPA2VideoProcessor'}(VJEPA2Config モデル)
プライベートモデルを使用する場合は、
token=Trueを渡す必要があります。
例:
>>> from transformers import AutoVideoProcessor
>>> # Download video processor from huggingface.co and cache.
>>> video_processor = AutoVideoProcessor.from_pretrained("llava-hf/llava-onevision-qwen2-0.5b-ov-hf")
>>> # If video processor files are in a directory (e.g. video processor was saved using *save_pretrained('./test/saved_model/')*)
>>> # video_processor = AutoVideoProcessor.from_pretrained("./test/saved_model/")register
< source >( config_classvideo_processor_class: type | None = Nonevideo_processor_classes: dict[str, type] | None = Noneexist_ok: bool = False )
パラメータ
- config_class (PreTrainedConfig) — 登録対象のモデルに対応する設定です。
- video_processor_class (BaseVideoProcessor) — 登録するビデオプロセッサ。
- video_processor_classes (
dict[str, type]、任意) — バックエンド名をビデオプロセッサクラスにマッピングする辞書です。カスタムバックエンドの登録を許可します。 例:{"pil": MyPilVideoProcessor, "torchvision": MyTorchvisionVideoProcessor, "custom": MyCustomVideoProcessor} - exist_ok (
bool、省略可、デフォルトはFalse) —Trueの場合、既存の登録を上書きすることを許可します。
このクラスに新しいビデオプロセッサを登録します。
AutoProcessor
これは汎用のプロセッサクラスであり、AutoProcessor.from_pretrained() クラスメソッドで生成された場合、ライブラリのプロセッサクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_pretrained
< source >( pretrained_model_name_or_path**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 次のいずれかです:- 文字列、huggingface.co上のモデルリポジトリにホストされている事前学習済みfeature_extractorのmodel id
save_pretrained()メソッドを使用してプロセッサのファイルを保存した ディレクトリ へのパス。例:./my_model_directory/
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルのフィーチャーエクストラクターをキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — feature extractor のファイルを(再)ダウンロードを強制し、キャッシュされたバージョンが存在する場合にそれを上書きするかどうかを指定します。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}.各リクエストでこれらのプロキシが使用されます。 - token (
strまたは bool、省略可) — リモートファイルの HTTP ベアラ認証に使用するトークン。Trueの場合、hf auth loginを実行した際に生成されたトークン(~/.huggingfaceに保存)を使用します。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - return_unused_kwargs (
bool、任意、デフォルトはFalse) —Falseの場合、この関数は最終的な feature extractor オブジェクトのみを返します。Trueの場合、この関数はTuple(feature_extractor, unused_kwargs)を返します。ここで unused_kwargs は、キーが feature extractor の属性でないキー/値のペアで構成される辞書です。つまり、feature_extractorの更新に使用されず、それ以外は無視されるkwargsの部分です。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - kwargs (
dict[str, Any], 任意) — kwargs 内の、feature extractor の属性であるキーの値は、読み込まれた値の上書きに使用されます。キーが feature extractor の属性でないキー/値ペアの動作は、return_unused_kwargsキーワード引数によって制御されます。
事前学習済みモデルの語彙から、ライブラリのプロセッサクラスのいずれかをインスタンス化します。
インスタンス化されるプロセッサクラスは、設定オブジェクトのmodel_typeプロパティに基づいて選択されます(引数として渡されるか、可能な場合はpretrained_model_name_or_pathから読み込まれます):
- aimv2 — CLIPProcessor (Aimv2Config モデル)
- align — AlignProcessor(AlignConfigモデル)
- altclip — AltCLIPProcessor(AltCLIPConfig モデル)
- aria — AriaProcessor(AriaConfigモデル)
- audioflamingo3 — AudioFlamingo3Processor(AudioFlamingo3Config モデル)
- aya_vision — AyaVisionProcessor(AyaVisionConfig モデル)
- bark — BarkProcessor (BarkConfig モデル)
- blip — BlipProcessor(BlipConfig モデル)
- blip-2 — Blip2Processor(Blip2Config モデル)
- bridgetower — BridgeTowerProcessor(BridgeTowerConfig モデル)
- bros — BrosProcessor(BrosConfig モデル)
- canary — CanaryProcessor(CanaryConfig モデル)
- chameleon — ChameleonProcessor(ChameleonConfigモデル)
- chinese_clip — ChineseCLIPProcessor(ChineseCLIPConfig モデル)
- clap — ClapProcessor(ClapConfig モデル)
- clip — CLIPProcessor(CLIPConfig モデル)
- clipseg — CLIPSegProcessor(CLIPSegConfig モデル)
- clvp — ClvpProcessor(ClvpConfig モデル)
- cohere2_vision — Cohere2VisionProcessor(Cohere2VisionConfig モデル)
- cohere_asr — CohereAsrProcessor(CohereAsrConfig モデル)
- cohere_compass — CohereCompassProcessor(CohereCompassConfig モデル)
- colmodernvbert — ColModernVBertProcessor(ColModernVBertConfig モデル)
- colpali — ColPaliProcessor(ColPaliConfig モデル)
- colqwen2 — ColQwen2Processor(ColQwen2Config モデル)
- cosmos3_edge — Cosmos3EdgeProcessor(Cosmos3EdgeConfig モデル)
- csm — CsmProcessor(CsmConfigモデル)
- deepseek_ocr2 — DeepseekOcr2Processor(DeepseekOcr2Config モデル)
- deepseek_vl — DeepseekVLProcessor(DeepseekVLConfig モデル)
- deepseek_vl_hybrid — DeepseekVLHybridProcessor(DeepseekVLHybridConfig モデル)
- dia — DiaProcessor(DiaConfig モデル)
- diffusion_gemma — Gemma4Processor(DiffusionGemmaConfig モデル)
- dots3_note — Dots3NoteProcessor(Dots3NoteConfigモデル)
- edgetam — Sam2Processor(EdgeTamConfig モデル)
- embedding_gemma2 — EmbeddingGemma2Processor(EmbeddingGemma2Config モデル)
- emu3 — Emu3Processor(Emu3Config モデル)
- ernie4_5_vl_moe — Ernie4_5_VLMoeProcessor(Ernie4_5_VLMoeConfig モデル)
- evolla — EvollaProcessor(EvollaConfigモデル)
- exaone4_5 — Exaone4_5_Processor(Exaone4_5_Config モデル)
- flava — FlavaProcessor(FlavaConfigモデル)
- florence2 — Florence2Processor(Florence2Config モデル)
- fun_asr_nano — FunAsrNanoProcessor(FunAsrNanoConfig モデル)
- fuyu — FuyuProcessor(FuyuConfig モデル)
- gemma3 — Gemma3Processor(Gemma3Config モデル)
- gemma3n — Gemma3nProcessor(Gemma3nConfig モデル)
- gemma4 — Gemma4Processor(Gemma4Config モデル)
- gemma4_unified — Gemma4UnifiedProcessor(Gemma4UnifiedConfig モデル)
- git — GitProcessor(GitConfig モデル)
- glm46v — Glm46VProcessor(Glm46VConfig モデル)
- glm4v — Glm4vProcessor(Glm4vConfig モデル)
- glm4v_moe — Glm4vProcessor(Glm4vMoeConfig モデル)
- glm5_next — Glm5NextProcessor(Glm5NextConfig モデル)
- glm_image — GlmImageProcessor(GlmImageConfig モデル)
- glmasr — GlmAsrProcessor(GlmAsrConfig モデル)
- got_ocr2 — GotOcr2Processor(GotOcr2Config モデル)
- gradium_phonon_tts — PocketTTSProcessor(GradiumPhononTTSConfig モデル)
- granite4_vision — Granite4VisionProcessor(Granite4VisionConfig モデル)
- granite_docling — GraniteDoclingProcessor(GraniteDoclingConfig モデル)
- granite_speech — GraniteSpeechProcessor(GraniteSpeechConfig モデル)
- granite_speech5_ctc — GraniteSpeech5Processor(GraniteSpeech5CTCConfig モデル)
- granite_speech_plus — GraniteSpeechProcessor(GraniteSpeechPlusConfig モデル)
- grounding-dino — GroundingDinoProcessor(GroundingDinoConfig モデル)
- groupvit — CLIPProcessor(GroupViTConfigモデル)
- higgs_audio_v2 — HiggsAudioV2Processor(HiggsAudioV2Config モデル)
- hubert — Wav2Vec2Processor(HubertConfigモデル)
- hunyuan_vl — HunYuanVLProcessor(HunYuanVLConfig モデル)
- hyperclovax_vision_v2 — HyperCLOVAXVisionV2Processor(HyperCLOVAXVisionV2Config モデル)
- idefics — IdeficsProcessor(IdeficsConfig モデル)
- idefics2 — Idefics2Processor(Idefics2Config モデル)
- idefics3 — Idefics3Processor(Idefics3Config モデル)
- inkling2 — Inkling2Processor(Inkling2Config モデル)
- inkling_mm_model — InklingProcessor(InklingConfig モデル)
- instructblip — InstructBlipProcessor(InstructBlipConfig モデル)
- instructblipvideo — InstructBlipVideoProcessor(InstructBlipVideoConfig モデル)
- internvl — InternVLProcessor(InternVLConfig モデル)
- janus — JanusProcessor(JanusConfig モデル)
- kimi_k25 — Kimi_K25Processor(Kimi_K25Config モデル)
- kyutai_speech_to_text — KyutaiSpeechToTextProcessor(KyutaiSpeechToTextConfig モデル)
- lasr_ctc — LasrProcessor(LasrCTCConfig モデル)
- lasr_encoder — LasrProcessor(LasrEncoderConfig モデル)
- layoutlmv2 — LayoutLMv2Processor(LayoutLMv2Config モデル)
- layoutlmv3 — LayoutLMv3Processor(LayoutLMv3Config モデル)
- layoutxlm — LayoutXLMProcessor(LayoutXLMConfig モデル)
- lfm2_vl — Lfm2VlProcessor(Lfm2VlConfig モデル)
- lfm3_vl — Lfm3VlProcessor(Lfm3VlConfig モデル)
- lighton_ocr — LightOnOcrProcessor(LightOnOcrConfig モデル)
- llama4 — Llama4Processor(Llama4Config モデル)
- llava — LlavaProcessor(LlavaConfig モデル)
- llava_next — LlavaNextProcessor(LlavaNextConfig モデル)
- llava_next_video — LlavaNextVideoProcessor(LlavaNextVideoConfig モデル)
- llava_onevision — LlavaOnevisionProcessor(LlavaOnevisionConfig モデル)
- markuplm — MarkupLMProcessor(MarkupLMConfig モデル)
- metaclip_2 — CLIPProcessor(MetaClip2Config モデル)
- mgp-str — MgpstrProcessor(MgpstrConfig モデル)
- minicpm_ocr — MiniCPMOcrProcessor(MiniCPMOcrConfig モデル)
- minicpmv4_6 — MiniCPMV4_6Processor(MiniCPMV4_6Config モデル)
- minicpmv4_7 — MiniCPMV4_7Processor(MiniCPMV4_7Config モデル)
- minimax_m3_vl — MiniMaxM3VLProcessor(MiniMaxM3VLConfig モデル)
- mistral3 — PixtralProcessor(Mistral3Config モデル)
- mistral_large4 — PixtralProcessor(MistralLarge4Config モデル)
- mllama — MllamaProcessor(MllamaConfig モデル)
- mm-grounding-dino — GroundingDinoProcessor(MMGroundingDinoConfig モデル)
- modernvbert — Idefics3Processor(ModernVBertConfig モデル)
- molmo2 — Molmo2Processor(Molmo2Config モデル)
- moonshine — Wav2Vec2Processor(MoonshineConfig モデル)
- moonshine_streaming — MoonshineStreamingProcessor(MoonshineStreamingConfig モデル)
- moss_transcribe_diarize — MossTranscribeDiarizeProcessor(MossTranscribeDiarizeConfig モデル)
- muse_glimmer — MuseGlimmerProcessor(MuseGlimmerConfig モデル)
- muse_spark — MuseSparkProcessor(MuseSparkConfig モデル)
- musicflamingo — MusicFlamingoProcessor(MusicFlamingoConfig モデル)
- musicgen — MusicgenProcessor(MusicgenConfig モデル)
- musicgen_melody — MusicgenMelodyProcessor(MusicgenMelodyConfig モデル)
- nemo_speechlm — Nemotron3_5TranscribeProcessor(Nemotron3_5TranscribeConfig モデル)
- nemotron3_5_asr — Nemotron3_5AsrProcessor(Nemotron3_5AsrConfig モデル)
- nemotron3_diarization — Nemotron3DiarizationProcessor(Nemotron3DiarizationConfig モデル)
- nemotron_asr_streaming — NemotronAsrStreamingProcessor (NemotronAsrStreamingConfig モデル)
- nemotron_h_omni — NemotronH_Omni_Reasoning_V3Processor(NemotronH_Omni_Reasoning_V3_Config モデル)
- neomme — NeoMMEProcessor (NeoMMEConfig モデル)
- nougat — NougatProcessor(NougatConfig モデル)
- omdet-turbo — OmDetTurboProcessor(OmDetTurboConfig モデル)
- oneformer — OneFormerProcessor(OneFormerConfig モデル)
- ovis2 — Ovis2Processor(Ovis2Config モデル)
- owlv2 — Owlv2Processor(Owlv2Config モデル)
- owlvit — OwlViTProcessor (OwlViTConfig モデル)
- paddleocr_vl — PaddleOCRVLProcessor(PaddleOCRVLConfig モデル)
- paligemma — PaliGemmaProcessor(PaliGemmaConfig モデル)
- parakeet_ctc — ParakeetProcessor(ParakeetCTCConfig モデル)
- parakeet_rnnt — ParakeetProcessor(ParakeetRNNTConfigモデル)
- parakeet_tdt — ParakeetProcessor(ParakeetTDTConfig モデル)
- pe_audio — PeAudioProcessor(PeAudioConfig モデル)
- pe_audio_video — PeAudioVideoProcessor(PeAudioVideoConfig モデル)
- pe_video — PeVideoProcessor(PeVideoConfig モデル)
- perception_lm — PerceptionLMProcessor(PerceptionLMConfig モデル)
- phi4_multimodal — Phi4MultimodalProcessor(Phi4MultimodalConfig モデル)
- pi0 — PI0Processor(PI0Config モデル)
- pix2struct — Pix2StructProcessor(Pix2StructConfig モデル)
- pixtral — PixtralProcessor(PixtralVisionConfig モデル)
- pocket_tts — PocketTTSProcessor(PocketTTSConfig モデル)
- pop2piano — Pop2PianoProcessor(Pop2PianoConfig モデル)
- pp_chart2table — PPChart2TableProcessor(PPChart2TableConfig モデル)
- pp_formulanet — PPFormulaNetProcessor(PPFormulaNetConfig モデル)
- qianfan_ocr — QianfanOCRProcessor(QianfanOCRConfig モデル)
- qwen2_5_omni — Qwen2_5OmniProcessor(Qwen2_5OmniConfig モデル)
- qwen2_5_vl — Qwen2_5_VLProcessor(Qwen2_5_VLConfig モデル)
- qwen2_audio — Qwen2AudioProcessor(Qwen2AudioConfig モデル)
- qwen2_vl — Qwen2VLProcessor(Qwen2VLConfig モデル)
- qwen3_5 — Qwen3VLProcessor(Qwen3_5Config モデル)
- qwen3_5_moe — Qwen3VLProcessor(Qwen3_5MoeConfig モデル)
- qwen3_asr — Qwen3ASRProcessor(Qwen3ASRConfig モデル)
- qwen3_omni_moe — Qwen3OmniMoeProcessor(Qwen3OmniMoeConfig モデル)
- qwen3_vl — Qwen3VLProcessor(Qwen3VLConfig モデル)
- qwen3_vl_moe — Qwen3VLProcessor(Qwen3VLMoeConfig モデル)
- qwen4_exp — Qwen3VLProcessor(Qwen4ExpConfig モデル)
- sam — SamProcessor(SamConfig モデル)
- sam2 — Sam2Processor(Sam2Config モデル)
- sam2_video — Sam2VideoProcessor(Sam2VideoConfig モデル)
- sam3 — Sam3Processor(Sam3Config モデル)
- sam3_lite_text — Sam3Processor (Sam3LiteTextConfig モデル)
- sam3_tracker — Sam3TrackerProcessor(Sam3TrackerConfig モデル)
- sam3_tracker_video — Sam3TrackerVideoProcessor(Sam3TrackerVideoConfig モデル)
- sam3_video — Sam3VideoProcessor (Sam3VideoConfig モデル)
- sam_hq — SamHQProcessor(SamHQConfig モデル)
- seamless_m4t — SeamlessM4TProcessor(SeamlessM4TConfig モデル)
- sew — Wav2Vec2Processor(SEWConfig モデル)
- sew-d — Wav2Vec2Processor(SEWDConfigモデル)
- shieldgemma2 — ShieldGemma2Processor(ShieldGemma2Config モデル)
- siglip — SiglipProcessor(SiglipConfig モデル)
- siglip2 — Siglip2Processor(Siglip2Config モデル)
- smolvlm — SmolVLMProcessor(SmolVLMConfig モデル)
- speech_to_text — Speech2TextProcessor(Speech2TextConfig モデル)
- speecht5 — SpeechT5Processor(SpeechT5Config モデル)
- step3p7 — Step3p7Processor(Step3p7Config モデル)
- t5gemma2 — Gemma3Processor(T5Gemma2Config モデル)
- t5gemma2_encoder — Gemma3Processor(T5Gemma2EncoderConfig モデル)
- tipsv2 — Tipsv2Processor(Tipsv2Config モデル)
- trocr — TrOCRProcessor(TrOCRConfig モデル)
- tvp — TvpProcessor (TvpConfigモデル)
- udop — UdopProcessor (UdopConfig モデル)
- ultravbert — Idefics3Processor(UltraVBertConfig モデル)
- unispeech — Wav2Vec2Processor(UniSpeechConfig モデル)
- unispeech-sat — Wav2Vec2Processor(UniSpeechSatConfig モデル)
- unlimited_ocr — UnlimitedOcrProcessor(UnlimitedOcrConfig モデル)
- vibevoice — VibeVoiceProcessor(VibeVoiceConfig モデル)
- vibevoice_asr — VibeVoiceAsrProcessor(VibeVoiceAsrConfig モデル)
- video_llama_3 — VideoLlama3Processor(VideoLlama3Config モデル)
- video_llava — VideoLlavaProcessor(VideoLlavaConfig モデル)
- videoprism — VideoPrismProcessor(VideoPrismConfigモデル)
- vilt — ViltProcessor(ViltConfig モデル)
- vipllava — LlavaProcessor(VipLlavaConfig モデル)
- vision-text-dual-encoder — VisionTextDualEncoderProcessor(VisionTextDualEncoderConfig モデル)
- voxtral — VoxtralProcessor(VoxtralConfigモデル)
- voxtral_realtime — VoxtralRealtimeProcessor(VoxtralRealtimeConfig モデル)
- wav2vec2 — Wav2Vec2Processor(Wav2Vec2Config モデル)
- wav2vec2-bert — Wav2Vec2BertProcessor (Wav2Vec2BertConfig モデル)
- wav2vec2-conformer — Wav2Vec2Processor(Wav2Vec2ConformerConfig モデル)
- wavlm — Wav2Vec2Processor(WavLMConfig モデル)
- whisper — WhisperProcessor(WhisperConfig モデル)
プライベートモデルを使用する場合は、
token=Trueを渡す必要があります。
例:
>>> from transformers import AutoProcessor
>>> # Download processor from huggingface.co and cache.
>>> processor = AutoProcessor.from_pretrained("facebook/wav2vec2-base-960h")
>>> # If processor files are in a directory (e.g. processor was saved using *save_pretrained('./test/saved_model/')*)
>>> # processor = AutoProcessor.from_pretrained("./test/saved_model/")register
< source >( config_classprocessor_classexist_ok = False )
パラメータ
- config_class (PreTrainedConfig) — 登録対象のモデルに対応する設定です。
- processor_class (ProcessorMixin) — 登録するプロセッサ。
このクラスに新しいプロセッサを登録します。
汎用モデルクラス
ベースモデルクラスを特定のヘッドなしでインスタンス化するために、以下のautoクラスが利用可能です。
AutoModel
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドを使用して作成された場合、ライブラリのベースモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- ASTConfig 設定クラス: ASTModel (ASTConfig モデル)
- AXK1Config 設定クラス: AXK1Model (AXK1Config モデル)
- AXK2Config 設定クラス: AXK2Model (AXK2Config モデル)
- AfmoeConfig 設定クラス: AfmoeModel (AfmoeConfig モデル)
- Aimv2Config 設定クラス: Aimv2Model (Aimv2Config モデル)
- Aimv2VisionConfig 設定クラス: Aimv2VisionModel (Aimv2VisionConfig モデル)
- AlbertConfig 設定クラス: AlbertModel (AlbertConfig モデル)
- AlignConfig 設定クラス: AlignModel (AlignConfig モデル)
- AltCLIPConfig 設定クラス: AltCLIPModel (AltCLIPConfig モデル)
- ApertusConfig 設定クラス: ApertusModel (ApertusConfig モデル)
- ArceeConfig 設定クラス: ArceeModel (ArceeConfig モデル)
- AriaConfig 設定クラス: AriaModel (AriaConfig モデル)
- AriaTextConfig 設定クラス: AriaTextModel (AriaTextConfig モデル)
- AudioFlamingo3Config 設定クラス: AudioFlamingo3Model (AudioFlamingo3Config モデル)
- AudioFlamingo3EncoderConfig 設定クラス: AudioFlamingo3Encoder (AudioFlamingo3EncoderConfig モデル)
- AutoformerConfig 設定クラス: AutoformerModel (AutoformerConfig モデル)
- AyaVisionConfig 設定クラス: AyaVisionModel (AyaVisionConfig モデル)
- BailingHybridConfig 設定クラス: BailingHybridModel (BailingHybridConfig モデル)
- BambaConfig 設定クラス: BambaModel (BambaConfig モデル)
- BarkConfig 設定クラス: BarkModel (BarkConfig モデル)
- BartConfig 設定クラス: BartModel (BartConfig モデル)
- BeamConfig 設定クラス: BeamModel (BeamConfig モデル)
- BeitConfig 設定クラス: BeitModel (BeitConfig モデル)
- BertConfig 設定クラス: BertModel (BertConfig モデル)
- BertGenerationConfig 設定クラス: BertGenerationEncoder (BertGenerationConfig モデル)
- BigBirdConfig 設定クラス: BigBirdModel (BigBirdConfig モデル)
- BigBirdPegasusConfig 設定クラス: BigBirdPegasusModel (BigBirdPegasusConfig モデル)
- BioGptConfig 設定クラス: BioGptModel (BioGptConfig モデル)
- BitConfig 設定クラス: BitModel(BitConfig モデル)
- BitNetConfig 設定クラス: BitNetModel (BitNetConfig モデル)
- BlenderbotConfig 設定クラス: BlenderbotModel (BlenderbotConfig モデル)
- BlenderbotSmallConfig 設定クラス: BlenderbotSmallModel (BlenderbotSmallConfig モデル)
- Blip2Config 設定クラス: Blip2Model (Blip2Config モデル)
- Blip2QFormerConfig 設定クラス: Blip2QFormerModel (Blip2QFormerConfig モデル)
- BlipConfig 設定クラス: BlipModel (BlipConfig モデル)
- BloomConfig 設定クラス: BloomModel (BloomConfig モデル)
- BltConfig 設定クラス: BltModel (BltConfig モデル)
- BridgeTowerConfig 設定クラス: BridgeTowerModel (BridgeTowerConfig モデル)
- BrosConfig 設定クラス: BrosModel (BrosConfig モデル)
- CLIPConfig 設定クラス: CLIPModel (CLIPConfig モデル)
- CLIPSegConfig 設定クラス: CLIPSegModel (CLIPSegConfig モデル)
- CLIPTextConfig 設定クラス: CLIPTextModel (CLIPTextConfig モデル)
- CLIPVisionConfig 設定クラス: CLIPVisionModel (CLIPVisionConfig モデル)
- CTRLConfig 設定クラス: CTRLModel (CTRLConfig モデル)
- CamembertConfig 設定クラス: CamembertModel (CamembertConfig モデル)
- CanaryConfig 設定クラス: CanaryModel (CanaryConfig モデル)
- CanineConfig 設定クラス: CanineModel (CanineConfig モデル)
- ChameleonConfig 設定クラス: ChameleonModel (ChameleonConfig モデル)
- ChineseCLIPConfig 設定クラス: ChineseCLIPModel (ChineseCLIPConfig モデル)
- ChineseCLIPVisionConfig 設定クラス: ChineseCLIPVisionModel (ChineseCLIPVisionConfig モデル)
- ClapConfig 設定クラス: ClapModel (ClapConfig モデル)
- ClvpConfig 設定クラス: ClvpModelForConditionalGeneration (ClvpConfig モデル)
- CodeGenConfig 設定クラス: CodeGenModel (CodeGenConfig モデル)
- Cohere2Config 設定クラス: Cohere2Model (Cohere2Config モデル)
- Cohere2MoeConfig 設定クラス: Cohere2MoeModel (Cohere2MoeConfig モデル)
- Cohere2VisionConfig 設定クラス: Cohere2VisionModel (Cohere2VisionConfig モデル)
- CohereAsrConfig 設定クラス: CohereAsrModel (CohereAsrConfig モデル)
- CohereCompassConfig 設定クラス: CohereCompassModel (CohereCompassConfig モデル)
- CohereCompassTextConfig 設定クラス: CohereCompassTextModel (CohereCompassTextConfig モデル)
- CohereCompassVisionConfig 設定クラス: CohereCompassVisionModel (CohereCompassVisionConfig モデル)
- CohereConfig 設定クラス: CohereModel (CohereConfig モデル)
- ConditionalDetrConfig 設定クラス: ConditionalDetrModel (ConditionalDetrConfig モデル)
- ConvBertConfig 設定クラス: ConvBertModel (ConvBertConfig モデル)
- ConvNextConfig 設定クラス: ConvNextModel (ConvNextConfig モデル)
- ConvNextV2Config 設定クラス: ConvNextV2Model (ConvNextV2Config モデル)
- Cosmos3EdgeConfig 設定クラス: Cosmos3EdgeModel (Cosmos3EdgeConfig モデル)
- Cosmos3EdgeTextConfig 設定クラス: Cosmos3EdgeTextModel (Cosmos3EdgeTextConfig モデル)
- Cosmos3EdgeVisionConfig 設定クラス: Cosmos3EdgeVisionModel (Cosmos3EdgeVisionConfig モデル)
- Cosmos3OmniConfig 設定クラス: Cosmos3OmniModel (Cosmos3OmniConfig モデル)
- CpmAntConfig 設定クラス: CpmAntModel (CpmAntConfig モデル)
- CsmConfig 設定クラス: CsmForConditionalGeneration (CsmConfig モデル)
- CvtConfig 設定クラス: CvtModel (CvtConfig モデル)
- CwmConfig 設定クラス: CwmModel (CwmConfig モデル)
- DFineConfig 設定クラス: DFineModel (DFineConfig モデル)
- DINOv3ConvNextConfig 設定クラス: DINOv3ConvNextModel (DINOv3ConvNextConfig モデル)
- DINOv3ViTConfig 設定クラス: DINOv3ViTModel (DINOv3ViTConfig モデル)
- DPRConfig 設定クラス: DPRQuestionEncoder (DPRConfig モデル)
- DPTConfig 設定クラス: DPTModel (DPTConfig モデル)
- DabDetrConfig 設定クラス: DabDetrModel (DabDetrConfig モデル)
- DacConfig 設定クラス: DacModel (DacConfig モデル)
- Data2VecAudioConfig 設定クラス: Data2VecAudioModel (Data2VecAudioConfig モデル)
- Data2VecTextConfig 設定クラス: Data2VecTextModel (Data2VecTextConfig モデル)
- Data2VecVisionConfig 設定クラス: Data2VecVisionModel (Data2VecVisionConfig モデル)
- DbrxConfig 設定クラス: DbrxModel (DbrxConfig モデル)
- DebertaConfig 設定クラス: DebertaModel (DebertaConfig モデル)
- DebertaV2Config 設定クラス: DebertaV2Model (DebertaV2Config モデル)
- DecisionTransformerConfig 設定クラス: DecisionTransformerModel (DecisionTransformerConfig モデル)
- DeepseekOcr2Config 設定クラス: DeepseekOcr2Model (DeepseekOcr2Config モデル)
- DeepseekV2Config 設定クラス: DeepseekV2Model (DeepseekV2Config モデル)
- DeepseekV32Config 設定クラス: DeepseekV32Model (DeepseekV32Config モデル)
- DeepseekV3Config 設定クラス: DeepseekV3Model (DeepseekV3Config モデル)
- DeepseekV4Config 設定クラス: DeepseekV4Model (DeepseekV4Config モデル)
- DeepseekVLConfig 設定クラス: DeepseekVLModel (DeepseekVLConfig モデル)
- DeepseekVLHybridConfig 設定クラス: DeepseekVLHybridModel (DeepseekVLHybridConfig モデル)
- DeformableDetrConfig 設定クラス: DeformableDetrModel (DeformableDetrConfig モデル)
- DeiTConfig 設定クラス: DeiTModel (DeiTConfig モデル)
- Deimv2Config 設定クラス: Deimv2Model (Deimv2Config モデル)
- DepthProConfig 設定クラス: DepthProModel (DepthProConfig モデル)
- DetrConfig 設定クラス: DetrModel (DetrConfig モデル)
- DiaConfig 設定クラス: DiaModel (DiaConfig モデル)
- DiffLlamaConfig 設定クラス: DiffLlamaModel (DiffLlamaConfig モデル)
- DiffusionGemmaConfig 設定クラス: DiffusionGemmaModel (DiffusionGemmaConfig モデル)
- DinatConfig 設定クラス: DinatModel (DinatConfig モデル)
- Dinov2Config 設定クラス: Dinov2Model (Dinov2Config モデル)
- Dinov2WithRegistersConfig 設定クラス: Dinov2WithRegistersModel (Dinov2WithRegistersConfig モデル)
- DistilBertConfig 設定クラス: DistilBertModel (DistilBertConfig モデル)
- DogeConfig 設定クラス: DogeModel (DogeConfig モデル)
- DonutSwinConfig 設定クラス: DonutSwinModel (DonutSwinConfig モデル)
- Dots1Config 設定クラス: Dots1Model (Dots1Config モデル)
- Dots3NoteConfig 設定クラス: Dots3NoteModel (Dots3NoteConfig モデル)
- Dots3NoteTextConfig 設定クラス: Dots3NoteTextModel (Dots3NoteTextConfig モデル)
- EdgeTamConfig 設定クラス: EdgeTamModel (EdgeTamConfig モデル)
- EdgeTamVideoConfig 設定クラス: EdgeTamVideoModel (EdgeTamVideoConfig モデル)
- EdgeTamVisionConfig 設定クラス: EdgeTamVisionModel (EdgeTamVisionConfig モデル)
- EfficientLoFTRConfig 設定クラス: EfficientLoFTRModel (EfficientLoFTRConfig モデル)
- EfficientNetConfig 設定クラス: EfficientNetModel (EfficientNetConfig モデル)
- ElectraConfig 設定クラス: ElectraModel (ElectraConfig モデル)
- EmbeddingGemma2Config 設定クラス: EmbeddingGemma2Model (EmbeddingGemma2Config モデル)
- EmbeddingGemma2TextConfig 設定クラス: EmbeddingGemma2TextModel (EmbeddingGemma2TextConfig モデル)
- Emu3Config 設定クラス: Emu3Model (Emu3Config モデル)
- EncodecConfig 設定クラス: EncodecModel (EncodecConfig モデル)
- Ernie4_5Config 設定クラス: Ernie4_5Model (Ernie4_5Config モデル)
- Ernie4_5_MoeConfig 設定クラス: Ernie4_5_MoeModel (Ernie4_5_MoeConfig モデル)
- Ernie4_5_VLMoeConfig 設定クラス: Ernie4_5_VLMoeModel (Ernie4_5_VLMoeConfig モデル)
- ErnieConfig 設定クラス: ErnieModel (ErnieConfig モデル)
- EsmConfig 設定クラス: EsmModel (EsmConfig モデル)
- EsmcConfig 設定クラス: EsmcModel (EsmcConfig モデル)
- EuroBertConfig 設定クラス: EuroBertModel (EuroBertConfig モデル)
- EvollaConfig 設定クラス: EvollaModel (EvollaConfig モデル)
- Exaone4Config 設定クラス: Exaone4Model (Exaone4Config モデル)
- Exaone4_5_Config 設定クラス: Exaone4_5_Model (Exaone4_5_Config モデル)
- Exaone4_5_VisionConfig 設定クラス: Exaone4_5_VisionModel (Exaone4_5_VisionConfig モデル)
- ExaoneMoeConfig 設定クラス: ExaoneMoeModel (ExaoneMoeConfig モデル)
- FNetConfig 設定クラス: FNetModel (FNetConfig モデル)
- FSMTConfig 設定クラス: FSMTModel (FSMTConfig モデル)
- FalconConfig 設定クラス: FalconModel (FalconConfig モデル)
- FalconH1Config 設定クラス: FalconH1Model (FalconH1Config モデル)
- FalconMambaConfig 設定クラス: FalconMambaModel (FalconMambaConfig モデル)
- FastSpeech2ConformerConfig 設定クラス: FastSpeech2ConformerModel (FastSpeech2ConformerConfig モデル)
- FastSpeech2ConformerWithHifiGanConfig 設定クラス: FastSpeech2ConformerWithHifiGan (FastSpeech2ConformerWithHifiGanConfig モデル)
- FastVlmConfig 設定クラス: FastVlmModel (FastVlmConfig モデル)
- FlaubertConfig 設定クラス: FlaubertModel (FlaubertConfig モデル)
- FlavaConfig 設定クラス: FlavaModel (FlavaConfig モデル)
- FlexOlmoConfig 設定クラス: FlexOlmoModel (FlexOlmoConfig モデル)
- FlintConfig 設定クラス: FlintModel (FlintConfig モデル)
- Florence2Config 設定クラス: Florence2Model (Florence2Config モデル)
- FocalNetConfig 設定クラス: FocalNetModel (FocalNetConfig モデル)
- FunAsrNanoConfig 設定クラス: FunAsrNanoModel (FunAsrNanoConfig モデル)
- FunAsrNanoEncoderConfig 設定クラス: FunAsrNanoEncoder (FunAsrNanoEncoderConfig モデル)
- FunnelConfig 設定クラス: FunnelModel または FunnelBaseModel (FunnelConfig モデル)
- FuyuConfig 設定クラス: FuyuModel (FuyuConfig モデル)
- GLPNConfig 設定クラス: GLPNModel (GLPNConfig モデル)
- GPT2Config 設定クラス: GPT2Model (GPT2Config モデル)
- GPTBigCodeConfig 設定クラス: GPTBigCodeModel (GPTBigCodeConfig モデル)
- GPTJConfig 設定クラス: GPTJModel (GPTJConfig モデル)
- GPTNeoConfig 設定クラス: GPTNeoModel (GPTNeoConfig モデル)
- GPTNeoXConfig 設定クラス: GPTNeoXModel (GPTNeoXConfig モデル)
- GPTNeoXJapaneseConfig 設定クラス: GPTNeoXJapaneseModel (GPTNeoXJapaneseConfig モデル)
- Gemma2Config 設定クラス: Gemma2Model (Gemma2Config モデル)
- Gemma3Config 設定クラス: Gemma3Model (Gemma3Config モデル)
- Gemma3TextConfig 設定クラス: Gemma3TextModel (Gemma3TextConfig モデル)
- Gemma3nAudioConfig 設定クラス:
Gemma3nAudioEncoder(Gemma3nAudioConfig モデル) - Gemma3nConfig 設定クラス: Gemma3nModel (Gemma3nConfig モデル)
- Gemma3nTextConfig 設定クラス: Gemma3nTextModel (Gemma3nTextConfig モデル)
- Gemma3nVisionConfig 設定クラス: TimmWrapperModel (Gemma3nVisionConfig モデル)
- Gemma4AudioConfig 設定クラス: Gemma4AudioModel (Gemma4AudioConfig モデル)
- Gemma4Config 設定クラス: Gemma4Model (Gemma4Config モデル)
- Gemma4TextConfig 設定クラス: Gemma4TextModel (Gemma4TextConfig モデル)
- Gemma4UnifiedConfig 設定クラス: Gemma4UnifiedModel (Gemma4UnifiedConfig モデル)
- Gemma4UnifiedTextConfig 設定クラス: Gemma4UnifiedTextModel (Gemma4UnifiedTextConfig モデル)
- Gemma4VisionConfig 設定クラス: Gemma4VisionModel (Gemma4VisionConfig モデル)
- GemmaConfig 設定クラス: GemmaModel (GemmaConfig モデル)
- GitConfig 設定クラス: GitModel (GitConfig モデル)
- Glm46VConfig 設定クラス: Glm46VModel (Glm46VConfig モデル)
- Glm4Config 設定クラス: Glm4Model (Glm4Config モデル)
- Glm4MoeConfig 設定クラス: Glm4MoeModel (Glm4MoeConfig モデル)
- Glm4MoeLiteConfig 設定クラス: Glm4MoeLiteModel (Glm4MoeLiteConfig モデル)
- Glm4vConfig 設定クラス: Glm4vModel (Glm4vConfig モデル)
- Glm4vMoeConfig 設定クラス: Glm4vMoeModel (Glm4vMoeConfig モデル)
- Glm4vMoeTextConfig 設定クラス: Glm4vMoeTextModel (Glm4vMoeTextConfig モデル)
- Glm4vMoeVisionConfig 設定クラス: Glm4vMoeVisionModel (Glm4vMoeVisionConfig モデル)
- Glm4vTextConfig 設定クラス: Glm4vTextModel (Glm4vTextConfig モデル)
- Glm4vVisionConfig 設定クラス: Glm4vVisionModel (Glm4vVisionConfig モデル)
- Glm5NextConfig 設定クラス: Glm5NextModel (Glm5NextConfig モデル)
- Glm5NextTextConfig 設定クラス: Glm5NextTextModel (Glm5NextTextConfig モデル)
- Glm5NextVisionConfig 設定クラス: Glm5NextVisionModel (Glm5NextVisionConfig モデル)
- GlmAsrConfig 設定クラス: GlmAsrModel (GlmAsrConfig モデル)
- GlmAsrEncoderConfig 設定クラス: GlmAsrEncoder (GlmAsrEncoderConfig モデル)
- GlmConfig 設定クラス: GlmModel (GlmConfig モデル)
- GlmImageConfig 設定クラス: GlmImageModel (GlmImageConfig モデル)
- GlmImageTextConfig 設定クラス: GlmImageTextModel (GlmImageTextConfig モデル)
- GlmImageVQVAEConfig 設定クラス: GlmImageVQVAE (GlmImageVQVAEConfig モデル)
- GlmImageVisionConfig 設定クラス: GlmImageVisionModel (GlmImageVisionConfig モデル)
- GlmMoeDsaConfig 設定クラス: GlmMoeDsaModel (GlmMoeDsaConfig モデル)
- GlmOcrConfig 設定クラス: GlmOcrModel (GlmOcrConfig モデル)
- GlmOcrTextConfig 設定クラス: GlmOcrTextModel (GlmOcrTextConfig モデル)
- GlmOcrVisionConfig 設定クラス: GlmOcrVisionModel (GlmOcrVisionConfig モデル)
- GlmgaConfig 設定クラス: Glm46VModel (GlmgaConfig モデル)
- GotOcr2Config 設定クラス: GotOcr2Model (GotOcr2Config モデル)
- GptOssConfig 設定クラス: GptOssModel (GptOssConfig モデル)
- GradiumPhononTTSConfig 設定クラス: GradiumPhononTTSModel (GradiumPhononTTSConfig モデル)
- Granite4VisionConfig 設定クラス: Granite4VisionModel (Granite4VisionConfig モデル)
- GraniteConfig 設定クラス: GraniteModel (GraniteConfig モデル)
- GraniteDoclingConfig 設定クラス: GraniteDoclingModel (GraniteDoclingConfig モデル)
- GraniteDoclingTextConfig 設定クラス: GraniteDoclingTextModel (GraniteDoclingTextConfig モデル)
- GraniteDoclingVisionConfig 設定クラス: GraniteDoclingVisionModel (GraniteDoclingVisionConfig モデル)
- GraniteMoeConfig 設定クラス: GraniteMoeModel (GraniteMoeConfig モデル)
- GraniteMoeHybridConfig 設定クラス: GraniteMoeHybridModel (GraniteMoeHybridConfig モデル)
- GraniteMoeSWAConfig 設定クラス: GraniteMoeSWAModel (GraniteMoeSWAConfig モデル)
- GraniteMoeSharedConfig 設定クラス: GraniteMoeSharedModel (GraniteMoeSharedConfig モデル)
- GraniteSWAConfig 設定クラス: GraniteSWAModel (GraniteSWAConfig モデル)
- GraniteSpeech5CTCConfig 設定クラス: GraniteSpeech5ForCTC (GraniteSpeech5CTCConfig モデル)
- GraniteSpeech5EncoderConfig 設定クラス: GraniteSpeech5Encoder (GraniteSpeech5EncoderConfig モデル)
- GraniteSpeechConfig 設定クラス: GraniteSpeechModel (GraniteSpeechConfig モデル)
- GraniteSpeechPlusConfig 設定クラス: GraniteSpeechPlusModel (GraniteSpeechPlusConfig モデル)
- GroundingDinoConfig 設定クラス: GroundingDinoModel (GroundingDinoConfig モデル)
- GroupViTConfig 設定クラス: GroupViTModel (GroupViTConfig モデル)
- GteConfig 設定クラス: GteModel (GteConfig モデル)
- HGNetV2Config 設定クラス: HGNetV2Backbone (HGNetV2Config モデル)
- HYV3Config 設定クラス: HYV3Model (HYV3Config モデル)
- HYV4Config 設定クラス: HYV4Model (HYV4Config モデル)
- HeliumConfig 設定クラス: HeliumModel (HeliumConfig モデル)
- HieraConfig 設定クラス: HieraModel (HieraConfig モデル)
- HiggsAudioV2Config 設定クラス: HiggsAudioV2ForConditionalGeneration (HiggsAudioV2Config モデル)
- HiggsAudioV2TokenizerConfig 設定クラス: HiggsAudioV2TokenizerModel (HiggsAudioV2TokenizerConfig モデル)
- HrmTextConfig 設定クラス: HrmTextModel (HrmTextConfig モデル)
- HubertConfig 設定クラス: HubertModel (HubertConfig モデル)
- HunYuanDenseV1Config 設定クラス: HunYuanDenseV1Model (HunYuanDenseV1Config モデル)
- HunYuanMoEV1Config 設定クラス: HunYuanMoEV1Model (HunYuanMoEV1Config モデル)
- HunYuanVLConfig 設定クラス: HunYuanVLModel (HunYuanVLConfig モデル)
- HunYuanVLTextConfig 設定クラス: HunYuanVLTextModel (HunYuanVLTextConfig モデル)
- HunYuanVLVisionConfig 設定クラス: HunYuanVLVisionTransformer (HunYuanVLVisionConfig モデル)
- HyperCLOVAXConfig 設定クラス: HyperCLOVAXModel (HyperCLOVAXConfig モデル)
- HyperCLOVAXVisionV2Config 設定クラス: HyperCLOVAXVisionV2Model (HyperCLOVAXVisionV2Config モデル)
- IBertConfig 設定クラス: IBertModel (IBertConfig モデル)
- IJepaConfig 設定クラス: IJepaModel (IJepaConfig モデル)
- IQuestCoderConfig 設定クラス: LlamaModel (IQuestCoderConfig モデル)
- Idefics2Config 設定クラス: Idefics2Model (Idefics2Config モデル)
- Idefics3Config 設定クラス: Idefics3Model (Idefics3Config モデル)
- Idefics3VisionConfig 設定クラス: Idefics3VisionTransformer (Idefics3VisionConfig モデル)
- IdeficsConfig 設定クラス: IdeficsModel (IdeficsConfig モデル)
- ImageGPTConfig 設定クラス: ImageGPTModel (ImageGPTConfig モデル)
- InformerConfig 設定クラス: InformerModel (InformerConfig モデル)
- Inkling2Config 設定クラス: Inkling2Model (Inkling2Config モデル)
- Inkling2TextConfig 設定クラス: Inkling2TextModel (Inkling2TextConfig モデル)
- Inkling2VisionConfig 設定クラス: Inkling2VisionModel (Inkling2VisionConfig モデル)
- InklingConfig 設定クラス: InklingModel (InklingConfig モデル)
- InklingTextConfig 設定クラス: InklingTextModel (InklingTextConfig モデル)
- InstructBlipConfig 設定クラス: InstructBlipModel (InstructBlipConfig モデル)
- InstructBlipVideoConfig 設定クラス: InstructBlipVideoModel (InstructBlipVideoConfig モデル)
- InternLM3Config 設定クラス: LlamaModel (InternLM3Config モデル)
- InternVLConfig 設定クラス: InternVLModel (InternVLConfig モデル)
- InternVLVisionConfig 設定クラス: InternVLVisionModel (InternVLVisionConfig モデル)
- Jais2Config 設定クラス: Jais2Model (Jais2Config モデル)
- JambaConfig 設定クラス: JambaModel (JambaConfig モデル)
- JanusConfig 設定クラス: JanusModel (JanusConfig モデル)
- JetMoeConfig 設定クラス: JetMoeModel (JetMoeConfig モデル)
- JinaEmbeddingsV3Config 設定クラス: JinaEmbeddingsV3Model (JinaEmbeddingsV3Config モデル)
- KimiLinearConfig 設定クラス: KimiLinearModel (KimiLinearConfig モデル)
- Kimi_K25Config 設定クラス: Kimi_K25Model (Kimi_K25Config モデル)
- Kimi_K25VisionConfig 設定クラス: Kimi_K25VisionModel (Kimi_K25VisionConfig モデル)
- Kolibri1Config 設定クラス: Kolibri1Model (Kolibri1Config モデル)
- Kosmos2Config 設定クラス: Kosmos2Model (Kosmos2Config モデル)
- Kosmos2_5Config 設定クラス: Kosmos2_5Model (Kosmos2_5Config モデル)
- KyutaiSpeechToTextConfig 設定クラス: KyutaiSpeechToTextModel (KyutaiSpeechToTextConfig モデル)
- LEDConfig 設定クラス: LEDModel (LEDConfig モデル)
- LagunaConfig 設定クラス: LagunaModel (LagunaConfig モデル)
- LasrCTCConfig 設定クラス: LasrForCTC (LasrCTCConfig モデル)
- LasrEncoderConfig 設定クラス: LasrEncoder (LasrEncoderConfig モデル)
- LayoutLMConfig 設定クラス: LayoutLMModel (LayoutLMConfig モデル)
- LayoutLMv2Config 設定クラス: LayoutLMv2Model (LayoutLMv2Config モデル)
- LayoutLMv3Config 設定クラス: LayoutLMv3Model (LayoutLMv3Config モデル)
- LevitConfig 設定クラス: LevitModel (LevitConfig モデル)
- Lfm2Config 設定クラス: Lfm2Model (Lfm2Config モデル)
- Lfm2MoeConfig 設定クラス: Lfm2MoeModel (Lfm2MoeConfig モデル)
- Lfm2VlConfig 設定クラス: Lfm2VlModel (Lfm2VlConfig モデル)
- Lfm3Config 設定クラス: Lfm3Model (Lfm3Config モデル)
- Lfm3VlConfig 設定クラス: Lfm3VlModel (Lfm3VlConfig モデル)
- Lfm3VlVisionConfig 設定クラス: Lfm3VlVisionModel (Lfm3VlVisionConfig モデル)
- LightGlueConfig 設定クラス: LightGlueForKeypointMatching (LightGlueConfig モデル)
- LightOnOcrConfig 設定クラス: LightOnOcrModel (LightOnOcrConfig モデル)
- LiltConfig 設定クラス: LiltModel (LiltConfig モデル)
- Llama4Config 設定クラス: Llama4ForConditionalGeneration (Llama4Config モデル)
- Llama4TextConfig 設定クラス: Llama4TextModel (Llama4TextConfig モデル)
- LlamaConfig 設定クラス: LlamaModel (LlamaConfig モデル)
- LlavaConfig 設定クラス: LlavaModel (LlavaConfig モデル)
- LlavaNextConfig 設定クラス: LlavaNextModel (LlavaNextConfig モデル)
- LlavaNextVideoConfig 設定クラス: LlavaNextVideoModel (LlavaNextVideoConfig モデル)
- LlavaOnevisionConfig 設定クラス: LlavaOnevisionModel (LlavaOnevisionConfig モデル)
- LongT5Config 設定クラス: LongT5Model (LongT5Config モデル)
- LongcatFlashConfig 設定クラス: LongcatFlashModel (LongcatFlashConfig モデル)
- LongformerConfig 設定クラス: LongformerModel (LongformerConfig モデル)
- LukeConfig 設定クラス: LukeModel (LukeConfig モデル)
- LwDetrConfig 設定クラス: LwDetrModel (LwDetrConfig モデル)
- LxmertConfig 設定クラス: LxmertModel (LxmertConfig モデル)
- M2M100Config 設定クラス: M2M100Model (M2M100Config モデル)
- MBartConfig 設定クラス: MBartModel (MBartConfig モデル)
- MLCDVisionConfig 設定クラス: MLCDVisionModel (MLCDVisionConfig モデル)
- MMGroundingDinoConfig 設定クラス: MMGroundingDinoModel (MMGroundingDinoConfig モデル)
- MPNetConfig 設定クラス: MPNetModel (MPNetConfig モデル)
- MT5Config 設定クラス: MT5Model (MT5Config モデル)
- Mamba2Config 設定クラス: Mamba2Model (Mamba2Config モデル)
- MambaConfig 設定クラス: MambaModel (MambaConfig モデル)
- MarianConfig 設定クラス: MarianModel (MarianConfig モデル)
- MarkupLMConfig 設定クラス: MarkupLMModel (MarkupLMConfig モデル)
- Mask2FormerConfig 設定クラス: Mask2FormerModel (Mask2FormerConfig モデル)
- MaskFormerConfig 設定クラス: MaskFormerModel (MaskFormerConfig モデル)
MaskFormerSwinConfig設定クラス:MaskFormerSwinModel(MaskFormerSwinConfig モデル)- MegatronBertConfig 設定クラス: MegatronBertModel (MegatronBertConfig モデル)
- MellumConfig 設定クラス: MellumModel (MellumConfig モデル)
- MetaClip2Config 設定クラス: MetaClip2Model (MetaClip2Config モデル)
- MgpstrConfig 設定クラス: MgpstrForSceneTextRecognition (MgpstrConfig モデル)
- MiMoV2FlashConfig 設定クラス: MiMoV2FlashModel (MiMoV2FlashConfig モデル)
- MimiConfig 設定クラス: MimiModel (MimiConfig モデル)
- MimiDecoderConfig 設定クラス: MimiDecoderModel (MimiDecoderConfig モデル)
- MimiEncoderConfig 設定クラス: MimiEncoderModel (MimiEncoderConfig モデル)
- MiniCPM3Config 設定クラス: MiniCPM3Model (MiniCPM3Config モデル)
- MiniCPMOcrConfig 設定クラス: MiniCPMOcrModel(MiniCPMOcrConfig モデル)
- MiniCPMOcrVisionConfig 設定クラス: MiniCPMOcrVisionModel (MiniCPMOcrVisionConfig モデル)
- MiniCPMV4_6Config 設定クラス: MiniCPMV4_6Model (MiniCPMV4_6Config モデル)
- MiniCPMV4_7Config 設定クラス: MiniCPMV4_7Model (MiniCPMV4_7Config モデル)
- MiniCPMV4_7VisionConfig 設定クラス: MiniCPMV4_7VisionModel (MiniCPMV4_7VisionConfig モデル)
- MiniMaxConfig 設定クラス: MiniMaxModel (MiniMaxConfig モデル)
- MiniMaxM2Config 設定クラス: MiniMaxM2Model (MiniMaxM2Config モデル)
- MiniMaxM3VLConfig 設定クラス: MiniMaxM3VLModel (MiniMaxM3VLConfig モデル)
- MiniMaxM3VLTextConfig 設定クラス: MiniMaxM3VLTextModel (MiniMaxM3VLTextConfig モデル)
- Ministral3Config 設定クラス: Ministral3Model (Ministral3Config モデル)
- MinistralConfig 設定クラス: MinistralModel (MinistralConfig モデル)
- Mistral3Config 設定クラス: Mistral3Model (Mistral3Config モデル)
- Mistral4Config 設定クラス: Mistral4Model(Mistral4Config モデル)
- MistralConfig 設定クラス: MistralModel (MistralConfig モデル)
- MistralLarge4Config 設定クラス: MistralLarge4Model (MistralLarge4Config モデル)
- MistralLarge4TextConfig 設定クラス: MistralLarge4TextModel (MistralLarge4TextConfig モデル)
- MixtralConfig 設定クラス: MixtralModel (MixtralConfig モデル)
- MllamaConfig 設定クラス: MllamaModel (MllamaConfig モデル)
- MobileBertConfig 設定クラス: MobileBertModel (MobileBertConfig モデル)
- MobileNetV1Config 設定クラス: MobileNetV1Model (MobileNetV1Config モデル)
- MobileNetV2Config 設定クラス: MobileNetV2Model (MobileNetV2Config モデル)
- MobileViTConfig 設定クラス: MobileViTModel (MobileViTConfig モデル)
- MobileViTV2Config 設定クラス: MobileViTV2Model (MobileViTV2Config モデル)
- ModernBertConfig 設定クラス: ModernBertModel (ModernBertConfig モデル)
- ModernBertDecoderConfig 設定クラス: ModernBertDecoderModel (ModernBertDecoderConfig モデル)
- ModernVBertConfig 設定クラス: ModernVBertModel (ModernVBertConfig モデル)
- Molmo2Config 設定クラス: Molmo2Model (Molmo2Config モデル)
- Molmo2TextConfig 設定クラス: Molmo2TextModel (Molmo2TextConfig モデル)
- MoonshineConfig 設定クラス: MoonshineModel (MoonshineConfig モデル)
- MoonshineStreamingConfig 設定クラス: MoonshineStreamingModel (MoonshineStreamingConfig モデル)
- MoshiConfig 設定クラス: MoshiModel (MoshiConfig モデル)
- MossTranscribeDiarizeAudioConfig 設定クラス: MossTranscribeDiarizeAudioModel (MossTranscribeDiarizeAudioConfig モデル)
- MossTranscribeDiarizeConfig 設定クラス: MossTranscribeDiarizeModel (MossTranscribeDiarizeConfig モデル)
- MptConfig 設定クラス: MptModel (MptConfig モデル)
- MraConfig 設定クラス: MraModel (MraConfig モデル)
- MuseGlimmerAssistantConfig 設定クラス: MuseGlimmerAssistantModel (MuseGlimmerAssistantConfig モデル)
- MuseGlimmerConfig 設定クラス: MuseGlimmerModel (MuseGlimmerConfig モデル)
- MuseGlimmerTextConfig 設定クラス: MuseGlimmerTextModel (MuseGlimmerTextConfig モデル)
- MuseGlimmerVisionConfig 設定クラス: MuseGlimmerVisionModel (MuseGlimmerVisionConfig モデル)
- MuseSparkAudioConfig 設定クラス: MuseSparkAudioModel (MuseSparkAudioConfig モデル)
- MuseSparkConfig 設定クラス: MuseSparkModel (MuseSparkConfig モデル)
- MuseSparkTextConfig 設定クラス: MuseSparkTextModel (MuseSparkTextConfig モデル)
- MuseSparkVisionConfig 設定クラス: MuseSparkVisionModel (MuseSparkVisionConfig モデル)
- MusicFlamingoConfig 設定クラス: MusicFlamingoModel (MusicFlamingoConfig モデル)
- MusicgenConfig 設定クラス: MusicgenModel (MusicgenConfig モデル)
- MusicgenMelodyConfig 設定クラス: MusicgenMelodyModel (MusicgenMelodyConfig モデル)
- MvpConfig 設定クラス: MvpModel (MvpConfig モデル)
- NanoChatConfig 設定クラス: NanoChatModel (NanoChatConfig モデル)
- Nemotron3DiarizationAudioConfig 設定クラス: Nemotron3DiarizationAudioModel (Nemotron3DiarizationAudioConfig モデル)
- Nemotron3DiarizationConfig 設定クラス: Nemotron3DiarizationModel (Nemotron3DiarizationConfig モデル)
- Nemotron3_5AsrConfig 設定クラス: Nemotron3_5AsrForRNNT (Nemotron3_5AsrConfig モデル)
- Nemotron3_5TranscribeAsrAudioConfig 設定クラス: Nemotron3_5TranscribeAsrAudioModel (Nemotron3_5TranscribeAsrAudioConfig モデル)
- Nemotron3_5TranscribeAudioConfig 設定クラス: Nemotron3_5TranscribeAudioModel (Nemotron3_5TranscribeAudioConfig モデル)
- Nemotron3_5TranscribeConfig 設定クラス: Nemotron3_5TranscribeModel (Nemotron3_5TranscribeConfig モデル)
- Nemotron3_5TranscribeWordTimestampConfig 設定クラス: Nemotron3_5TranscribeWordTimestampModel (Nemotron3_5TranscribeWordTimestampConfig モデル)
- NemotronAsrStreamingConfig 設定クラス: NemotronAsrStreamingForRNNT (NemotronAsrStreamingConfig モデル)
- NemotronAsrStreamingEncoderConfig 設定クラス: NemotronAsrStreamingEncoder (NemotronAsrStreamingEncoderConfig モデル)
- NemotronConfig 設定クラス: NemotronModel (NemotronConfig モデル)
- NemotronHConfig 設定クラス: NemotronHModel (NemotronHConfig モデル)
- NeoMMEConfig 設定クラス: NeoMMEModel (NeoMMEConfig モデル)
- NeuCodecConfig 設定クラス: NeuCodecModel (NeuCodecConfig モデル)
- NllbMoeConfig 設定クラス: NllbMoeModel (NllbMoeConfig モデル)
- NomicBertConfig 設定クラス: NomicBertModel (NomicBertConfig モデル)
- NystromformerConfig 設定クラス: NystromformerModel (NystromformerConfig モデル)
- OPTConfig 設定クラス: OPTModel (OPTConfig モデル)
- Olmo2Config 設定クラス: Olmo2Model (Olmo2Config モデル)
- Olmo3Config 設定クラス: Olmo3Model(Olmo3Config モデル)
- OlmoConfig 設定クラス: OlmoModel (OlmoConfig モデル)
- OlmoHybridConfig 設定クラス: OlmoHybridModel (OlmoHybridConfig モデル)
- OlmoeConfig 設定クラス: OlmoeModel (OlmoeConfig モデル)
- OmDetTurboConfig 設定クラス: OmDetTurboForObjectDetection (OmDetTurboConfig モデル)
- OneFormerConfig 設定クラス: OneFormerModel (OneFormerConfig モデル)
- OpenAIGPTConfig 設定クラス: OpenAIGPTModel (OpenAIGPTConfig モデル)
- OpenAIPrivacyFilterConfig 設定クラス: OpenAIPrivacyFilterModel (OpenAIPrivacyFilterConfig モデル)
- Ovis2Config 設定クラス: Ovis2Model (Ovis2Config モデル)
- OwlViTConfig 設定クラス: OwlViTModel (OwlViTConfig モデル)
- Owlv2Config 設定クラス: Owlv2Model (Owlv2Config モデル)
- PI0Config 設定クラス: PI0Model (PI0Config モデル)
- PLBartConfig 設定クラス: PLBartModel (PLBartConfig モデル)
- PPDocLayoutV3Config 設定クラス: PPDocLayoutV3Model (PPDocLayoutV3Config モデル)
- PPDocLayoutV4Config 設定クラス: PPDocLayoutV4Model (PPDocLayoutV4Config モデル)
- PPOCRV5MobileRecConfig 設定クラス: PPOCRV5MobileRecModel (PPOCRV5MobileRecConfig モデル)
- PPOCRV5ServerRecConfig 設定クラス: PPOCRV5ServerRecModel (PPOCRV5ServerRecConfig モデル)
- PPOCRV6MediumDetConfig 設定クラス: PPOCRV6MediumDetModel (PPOCRV6MediumDetConfig モデル)
- PPOCRV6SmallDetConfig 設定クラス: PPOCRV6SmallDetModel (PPOCRV6SmallDetConfig モデル)
- PPOCRV6SmallRecConfig 設定クラス: PPOCRV6SmallRecModel (PPOCRV6SmallRecConfig モデル)
- PPOCRV6TinyRecConfig 設定クラス: PPOCRV6TinyRecModel (PPOCRV6TinyRecConfig モデル)
- PaddleOCRVLConfig 設定クラス: PaddleOCRVLModel (PaddleOCRVLConfig モデル)
- PaliGemmaConfig 設定クラス: PaliGemmaModel (PaliGemmaConfig モデル)
- ParakeetCTCConfig 設定クラス: ParakeetForCTC (ParakeetCTCConfig モデル)
- ParakeetEncoderConfig 設定クラス: ParakeetEncoder (ParakeetEncoderConfig モデル)
- ParakeetRNNTConfig 設定クラス: ParakeetForRNNT (ParakeetRNNTConfig モデル)
- ParakeetTDTConfig 設定クラス: ParakeetForTDT (ParakeetTDTConfig モデル)
- PatchTSMixerConfig 設定クラス: PatchTSMixerModel (PatchTSMixerConfig モデル)
- PatchTSTConfig 設定クラス: PatchTSTModel (PatchTSTConfig モデル)
- PeAudioConfig 設定クラス: PeAudioModel (PeAudioConfig モデル)
- PeAudioEncoderConfig 設定クラス: PeAudioEncoder (PeAudioEncoderConfig モデル)
- PeAudioVideoConfig 設定クラス: PeAudioVideoModel(PeAudioVideoConfig モデル)
- PeAudioVideoEncoderConfig 設定クラス: PeAudioVideoEncoder(PeAudioVideoEncoderConfig モデル)
- PeVideoConfig 設定クラス: PeVideoModel (PeVideoConfig モデル)
- PeVideoEncoderConfig 設定クラス: PeVideoEncoder (PeVideoEncoderConfig モデル)
- PegasusConfig 設定クラス: PegasusModel (PegasusConfig モデル)
- PegasusXConfig 設定クラス: PegasusXModel (PegasusXConfig モデル)
- PerceiverConfig 設定クラス: PerceiverModel (PerceiverConfig モデル)
- PerceptionLMConfig 設定クラス: PerceptionLMModel (PerceptionLMConfig モデル)
- PersimmonConfig 設定クラス: PersimmonModel (PersimmonConfig モデル)
- Phi3Config 設定クラス: Phi3Model (Phi3Config モデル)
- Phi4MultimodalConfig 設定クラス: Phi4MultimodalModel (Phi4MultimodalConfig モデル)
- PhiConfig 設定クラス: PhiModel (PhiConfig モデル)
- PhimoeConfig 設定クラス: PhimoeModel (PhimoeConfig モデル)
- PixioConfig 設定クラス: PixioModel (PixioConfig モデル)
- PixtralVisionConfig 設定クラス: PixtralVisionModel (PixtralVisionConfig モデル)
- PocketTTSAudioTokenizerConfig 設定クラス: PocketTTSAudioTokenizerModel (PocketTTSAudioTokenizerConfig モデル)
- PocketTTSAudioTokenizerDecoderConfig 設定クラス: PocketTTSAudioTokenizerDecoderModel (PocketTTSAudioTokenizerDecoderConfig モデル)
- PocketTTSAudioTokenizerEncoderConfig 設定クラス: PocketTTSAudioTokenizerEncoderModel (PocketTTSAudioTokenizerEncoderConfig モデル)
- PocketTTSConfig 設定クラス: PocketTTSModel (PocketTTSConfig モデル)
- PoolFormerConfig 設定クラス: PoolFormerModel (PoolFormerConfig モデル)
- ProphetNetConfig 設定クラス: ProphetNetModel (ProphetNetConfig モデル)
- PvtConfig 設定クラス: PvtModel (PvtConfig モデル)
- PvtV2Config 設定クラス: PvtV2Model (PvtV2Config モデル)
- QianfanOCRConfig 設定クラス: QianfanOCRModel (QianfanOCRConfig モデル)
- QianfanOCRVisionConfig 設定クラス: QianfanOCRVisionModel (QianfanOCRVisionConfig モデル)
- Qwen2AudioConfig 設定クラス: Qwen2AudioModel (Qwen2AudioConfig モデル)
- Qwen2AudioEncoderConfig 設定クラス: Qwen2AudioEncoder (Qwen2AudioEncoderConfig モデル)
- Qwen2Config 設定クラス: Qwen2Model (Qwen2Config モデル)
- Qwen2MoeConfig 設定クラス: Qwen2MoeModel (Qwen2MoeConfig モデル)
- Qwen2VLConfig 設定クラス: Qwen2VLModel (Qwen2VLConfig モデル)
- Qwen2VLTextConfig 設定クラス: Qwen2VLTextModel (Qwen2VLTextConfig モデル)
- Qwen2_5OmniThinkerConfig 設定クラス: Qwen2_5OmniThinkerModel (Qwen2_5OmniThinkerConfig モデル)
- Qwen2_5_VLConfig 設定クラス: Qwen2_5_VLModel (Qwen2_5_VLConfig モデル)
- Qwen2_5_VLTextConfig 設定クラス: Qwen2_5_VLTextModel (Qwen2_5_VLTextConfig モデル)
- Qwen2_5_VLVisionConfig 設定クラス: Qwen2_5_VisionTransformerPretrainedModel (Qwen2_5_VLVisionConfig モデル)
- Qwen3ASRConfig 設定クラス: Qwen3ASRModel (Qwen3ASRConfig モデル)
- Qwen3ASREncoderConfig 設定クラス: Qwen3ASREncoder (Qwen3ASREncoderConfig モデル)
- Qwen3Config 設定クラス: Qwen3Model (Qwen3Config モデル)
- Qwen3MoeConfig 設定クラス: Qwen3MoeModel (Qwen3MoeConfig モデル)
- Qwen3NextConfig 設定クラス: Qwen3NextModel (Qwen3NextConfig モデル)
- Qwen3OmniMoeThinkerConfig 設定クラス: Qwen3OmniMoeThinkerModel(Qwen3OmniMoeThinkerConfig モデル)
- Qwen3VLConfig 設定クラス: Qwen3VLModel (Qwen3VLConfig モデル)
- Qwen3VLMoeConfig 設定クラス: Qwen3VLMoeModel (Qwen3VLMoeConfig モデル)
- Qwen3VLMoeTextConfig 設定クラス: Qwen3VLMoeTextModel (Qwen3VLMoeTextConfig モデル)
- Qwen3VLMoeVisionConfig 設定クラス: Qwen3VLMoeVisionModel (Qwen3VLMoeVisionConfig モデル)
- Qwen3VLTextConfig 設定クラス: Qwen3VLTextModel (Qwen3VLTextConfig モデル)
- Qwen3VLVisionConfig 設定クラス: Qwen3VLVisionModel (Qwen3VLVisionConfig モデル)
- Qwen3_5Config 設定クラス: Qwen3_5Model (Qwen3_5Config モデル)
- Qwen3_5MoeConfig 設定クラス: Qwen3_5MoeModel (Qwen3_5MoeConfig モデル)
- Qwen3_5MoeTextConfig 設定クラス: Qwen3_5MoeTextModel (Qwen3_5MoeTextConfig モデル)
- Qwen3_5MoeVisionConfig 設定クラス: Qwen3_5MoeVisionModel (Qwen3_5MoeVisionConfig モデル)
- Qwen3_5TextConfig 設定クラス: Qwen3_5TextModel (Qwen3_5TextConfig モデル)
- Qwen3_5VisionConfig 設定クラス: Qwen3_5VisionModel (Qwen3_5VisionConfig モデル)
- Qwen4ExpConfig 設定クラス: Qwen4ExpModel (Qwen4ExpConfig モデル)
- Qwen4ExpTextConfig 設定クラス: Qwen4ExpTextModel (Qwen4ExpTextConfig モデル)
- Qwen4ExpVisionConfig 設定クラス: Qwen4ExpVisionModel (Qwen4ExpVisionConfig モデル)
- RTDetrConfig 設定クラス: RTDetrModel (RTDetrConfig モデル)
- RTDetrV2Config 設定クラス: RTDetrV2Model (RTDetrV2Config モデル)
- RadioConfig 設定クラス: RadioModel (RadioConfig モデル)
- RecurrentGemmaConfig 設定クラス: RecurrentGemmaModel (RecurrentGemmaConfig モデル)
- ReformerConfig 設定クラス: ReformerModel (ReformerConfig モデル)
- RegNetConfig 設定クラス: RegNetModel (RegNetConfig モデル)
- RemBertConfig 設定クラス: RemBertModel (RemBertConfig モデル)
- ResNetConfig 設定クラス: ResNetModel (ResNetConfig モデル)
- RfDetrConfig 設定クラス: RfDetrModel (RfDetrConfig モデル)
- RoCBertConfig 設定クラス: RoCBertModel (RoCBertConfig モデル)
- RoFormerConfig 設定クラス: RoFormerModel (RoFormerConfig モデル)
- RobertaConfig 設定クラス: RobertaModel (RobertaConfig モデル)
- RobertaPreLayerNormConfig 設定クラス: RobertaPreLayerNormModel (RobertaPreLayerNormConfig モデル)
- RwkvConfig 設定クラス: RwkvModel (RwkvConfig モデル)
- SEWConfig 設定クラス: SEWModel (SEWConfig モデル)
- SEWDConfig 設定クラス: SEWDModel (SEWDConfig モデル)
- Sam2Config 設定クラス: Sam2Model (Sam2Config モデル)
- Sam2HieraDetConfig 設定クラス: Sam2HieraDetModel (Sam2HieraDetConfig モデル)
- Sam2VideoConfig 設定クラス: Sam2VideoModel (Sam2VideoConfig モデル)
- Sam2VisionConfig 設定クラス: Sam2VisionModel (Sam2VisionConfig モデル)
- Sam3Config 設定クラス: Sam3Model (Sam3Config モデル)
- Sam3LiteTextConfig 設定クラス: Sam3LiteTextModel (Sam3LiteTextConfig モデル)
- Sam3LiteTextTextConfig 設定クラス: Sam3LiteTextTextModel (Sam3LiteTextTextConfig モデル)
- Sam3TrackerConfig 設定クラス: Sam3TrackerModel (Sam3TrackerConfig モデル)
- Sam3TrackerVideoConfig 設定クラス: Sam3TrackerVideoModel (Sam3TrackerVideoConfig モデル)
- Sam3ViTConfig 設定クラス: Sam3ViTModel (Sam3ViTConfig モデル)
- Sam3VideoConfig 設定クラス: Sam3VideoModel (Sam3VideoConfig モデル)
- Sam3VisionConfig 設定クラス: Sam3VisionModel (Sam3VisionConfig モデル)
- SamConfig 設定クラス: SamModel (SamConfig モデル)
- SamHQConfig 設定クラス: SamHQModel (SamHQConfig モデル)
- SamHQVisionConfig 設定クラス: SamHQVisionModel (SamHQVisionConfig モデル)
- SamVisionConfig 設定クラス: SamVisionModel (SamVisionConfig モデル)
- Sapiens2Config 設定クラス: Sapiens2Model (Sapiens2Config モデル)
- SeamlessM4TConfig 設定クラス: SeamlessM4TModel (SeamlessM4TConfig モデル)
- SeamlessM4Tv2Config 設定クラス: SeamlessM4Tv2Model (SeamlessM4Tv2Config モデル)
- SeedOssConfig 設定クラス: SeedOssModel (SeedOssConfig モデル)
- SegGptConfig 設定クラス: SegGptModel (SegGptConfig モデル)
- SegformerConfig 設定クラス: SegformerModel (SegformerConfig モデル)
- Siglip2Config 設定クラス: Siglip2Model (Siglip2Config モデル)
- Siglip2VisionConfig 設定クラス: Siglip2VisionModel (Siglip2VisionConfig モデル)
- SiglipConfig 設定クラス: SiglipModel (SiglipConfig モデル)
- SiglipVisionConfig 設定クラス: SiglipVisionModel (SiglipVisionConfig モデル)
- SmolLM3Config 設定クラス: SmolLM3Model (SmolLM3Config モデル)
- SmolVLMConfig 設定クラス: SmolVLMModel (SmolVLMConfig モデル)
- SmolVLMVisionConfig 設定クラス: SmolVLMVisionTransformer (SmolVLMVisionConfig モデル)
- SolarOpenConfig 設定クラス: SolarOpenModel (SolarOpenConfig モデル)
- Speech2TextConfig 設定クラス: Speech2TextModel (Speech2TextConfig モデル)
- SpeechT5Config 設定クラス: SpeechT5Model (SpeechT5Config モデル)
- SplinterConfig 設定クラス: SplinterModel (SplinterConfig モデル)
- SqueezeBertConfig 設定クラス: SqueezeBertModel (SqueezeBertConfig モデル)
- StableLmConfig 設定クラス: StableLmModel (StableLmConfig モデル)
- Starcoder2Config 設定クラス: Starcoder2Model (Starcoder2Config モデル)
- Step3p7Config 設定クラス: Step3p7Model (Step3p7Config モデル)
- SwiftFormerConfig 設定クラス: SwiftFormerModel (SwiftFormerConfig モデル)
- Swin2SRConfig 設定クラス: Swin2SRModel (Swin2SRConfig モデル)
- SwinConfig 設定クラス: SwinModel (SwinConfig モデル)
- Swinv2Config 設定クラス: Swinv2Model (Swinv2Config モデル)
- SwitchTransformersConfig 設定クラス: SwitchTransformersModel (SwitchTransformersConfig モデル)
- T5Config 設定クラス: T5Model (T5Config モデル)
- T5Gemma2Config 設定クラス: T5Gemma2Model (T5Gemma2Config モデル)
- T5Gemma2EncoderConfig 設定クラス:
T5Gemma2Encoder(T5Gemma2EncoderConfig モデル) - T5GemmaConfig 設定クラス: T5GemmaModel (T5GemmaConfig モデル)
- TableTransformerConfig 設定クラス: TableTransformerModel (TableTransformerConfig モデル)
- TapasConfig 設定クラス: TapasModel (TapasConfig モデル)
- TextNetConfig 設定クラス: TextNetModel (TextNetConfig モデル)
- TimeSeriesTransformerConfig 設定クラス: TimeSeriesTransformerModel (TimeSeriesTransformerConfig モデル)
- TimesFm2_5Config 設定クラス: TimesFm2_5Model (TimesFm2_5Config モデル)
- TimesFmConfig 設定クラス: TimesFmModel (TimesFmConfig モデル)
- TimesformerConfig 設定クラス: TimesformerModel (TimesformerConfig モデル)
- TimmBackboneConfig 設定クラス: TimmBackbone (TimmBackboneConfig モデル)
- TimmWrapperConfig 設定クラス: TimmWrapperModel (TimmWrapperConfig モデル)
- Tipsv2Config 設定クラス: Tipsv2Model (Tipsv2Config モデル)
- Tipsv2TextConfig 設定クラス: Tipsv2TextModel (Tipsv2TextConfig モデル)
- Tipsv2VisionConfig 設定クラス: Tipsv2VisionModel (Tipsv2VisionConfig モデル)
- TvpConfig 設定クラス: TvpModel (TvpConfig モデル)
- UMT5Config 設定クラス: UMT5Model (UMT5Config モデル)
- UVDocConfig 設定クラス: UVDocModel (UVDocConfig モデル)
- UdopConfig 設定クラス: UdopModel (UdopConfig モデル)
- UltraVBertConfig 設定クラス: UltraVBertModel (UltraVBertConfig モデル)
- UltrabertConfig 設定クラス: UltrabertModel (UltrabertConfig モデル)
- UniSpeechConfig 設定クラス: UniSpeechModel (UniSpeechConfig モデル)
- UniSpeechSatConfig 設定クラス: UniSpeechSatModel (UniSpeechSatConfig モデル)
- UnivNetConfig 設定クラス: UnivNetModel (UnivNetConfig モデル)
- UnlimitedOcrConfig 設定クラス: UnlimitedOcrModel (UnlimitedOcrConfig モデル)
- VJEPA2Config 設定クラス: VJEPA2Model (VJEPA2Config モデル)
- VaultGemmaConfig 設定クラス: VaultGemmaModel (VaultGemmaConfig モデル)
- ViTConfig 設定クラス: ViTModel (ViTConfig モデル)
- ViTMAEConfig 設定クラス: ViTMAEModel (ViTMAEConfig モデル)
- ViTMSNConfig 設定クラス: ViTMSNModel (ViTMSNConfig モデル)
- VibeVoiceAcousticTokenizerConfig 設定クラス: VibeVoiceAcousticTokenizerModel (VibeVoiceAcousticTokenizerConfig モデル)
- VibeVoiceAcousticTokenizerDecoderConfig 設定クラス: VibeVoiceAcousticTokenizerDecoderModel (VibeVoiceAcousticTokenizerDecoderConfig モデル)
- VibeVoiceAcousticTokenizerEncoderConfig 設定クラス: VibeVoiceAcousticTokenizerEncoderModel (VibeVoiceAcousticTokenizerEncoderConfig モデル)
- VibeVoiceAsrConfig 設定クラス: VibeVoiceAsrModel (VibeVoiceAsrConfig モデル)
- VibeVoiceConfig 設定クラス: VibeVoiceForConditionalGeneration (VibeVoiceConfig モデル)
- VideoLlama3Config 設定クラス: VideoLlama3Model (VideoLlama3Config モデル)
- VideoLlama3VisionConfig 設定クラス: VideoLlama3VisionModel (VideoLlama3VisionConfig モデル)
- VideoLlavaConfig 設定クラス: VideoLlavaModel (VideoLlavaConfig モデル)
- VideoMAEConfig 設定クラス: VideoMAEModel (VideoMAEConfig モデル)
- VideoPrismConfig 設定クラス: VideoPrismClipModel (VideoPrismConfig モデル)
- VideoPrismVisionConfig 設定クラス: VideoPrismVisionModel (VideoPrismVisionConfig モデル)
- ViltConfig 設定クラス: ViltModel (ViltConfig モデル)
- VipLlavaConfig 設定クラス: VipLlavaModel (VipLlavaConfig モデル)
- VisionTextDualEncoderConfig 設定クラス: VisionTextDualEncoderModel (VisionTextDualEncoderConfig モデル)
- VisualBertConfig 設定クラス: VisualBertModel (VisualBertConfig モデル)
- VitDetConfig 設定クラス: VitDetModel (VitDetConfig モデル)
- VitsConfig 設定クラス: VitsModel (VitsConfig モデル)
- VivitConfig 設定クラス: VivitModel (VivitConfig モデル)
- VoxtralConfig 設定クラス: VoxtralModel (VoxtralConfig モデル)
- VoxtralEncoderConfig 設定クラス: VoxtralEncoder (VoxtralEncoderConfig モデル)
- VoxtralRealtimeConfig 設定クラス: VoxtralRealtimeModel (VoxtralRealtimeConfig モデル)
- VoxtralRealtimeEncoderConfig 設定クラス: VoxtralRealtimeEncoder (VoxtralRealtimeEncoderConfig モデル)
- VoxtralRealtimeTextConfig 設定クラス:
VoxtralRealtimeTextModel(VoxtralRealtimeTextConfig モデル) - Wav2Vec2BertConfig 設定クラス: Wav2Vec2BertModel (Wav2Vec2BertConfig モデル)
- Wav2Vec2Config 設定クラス: Wav2Vec2Model (Wav2Vec2Config モデル)
- Wav2Vec2ConformerConfig 設定クラス: Wav2Vec2ConformerModel (Wav2Vec2ConformerConfig モデル)
- WavLMConfig 設定クラス: WavLMModel (WavLMConfig モデル)
- WhisperConfig 設定クラス: WhisperModel (WhisperConfig モデル)
- XCLIPConfig 設定クラス: XCLIPModel (XCLIPConfig モデル)
- XGLMConfig 設定クラス: XGLMModel (XGLMConfig モデル)
- XLMConfig 設定クラス: XLMModel (XLMConfig モデル)
- XLMRobertaConfig 設定クラス: XLMRobertaModel (XLMRobertaConfig モデル)
- XLMRobertaXLConfig 設定クラス: XLMRobertaXLModel (XLMRobertaXLConfig モデル)
- XLNetConfig 設定クラス: XLNetModel (XLNetConfig モデル)
- Xcodec2Config 設定クラス: Xcodec2Model (Xcodec2Config モデル)
- XcodecConfig 設定クラス: XcodecModel (XcodecConfig モデル)
- XmodConfig 設定クラス: XmodModel (XmodConfig モデル)
- YolosConfig 設定クラス: YolosModel (YolosConfig モデル)
- YosoConfig 設定クラス: YosoModel (YosoConfig モデル)
- YoutuConfig 設定クラス: YoutuModel (YoutuConfig モデル)
- Zamba2Config 設定クラス: Zamba2Model (Zamba2Config モデル)
- ZambaConfig 設定クラス: ZambaModel (ZambaConfig モデル)
- ZayaConfig 設定クラス: ZayaModel (ZayaConfig モデル)
- xLSTMConfig 設定クラス: xLSTMModel (xLSTMConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのベースモデルクラスのいずれかをインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
事前学習済みモデルから、ライブラリのベースモデルクラスのいずれかをインスタンス化します。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- afmoe — AfmoeModel(AfmoeConfig モデル)
- aimv2 — Aimv2Model (Aimv2Config モデル)
- aimv2_vision_model — Aimv2VisionModel(Aimv2VisionConfig モデル)
- albert — AlbertModel(AlbertConfig モデル)
- align — AlignModel(AlignConfig モデル)
- altclip — AltCLIPModel(AltCLIPConfig モデル)
- apertus — ApertusModel(ApertusConfig モデル)
- arcee — ArceeModel(ArceeConfigモデル)
- aria — AriaModel(AriaConfigモデル)
- aria_text — AriaTextModel(AriaTextConfig モデル)
- audio-spectrogram-transformer — ASTModel(ASTConfigモデル)
- audioflamingo3 — AudioFlamingo3Model (AudioFlamingo3Config モデル)
- audioflamingo3_encoder — AudioFlamingo3Encoder(AudioFlamingo3EncoderConfig モデル)
- autoformer — AutoformerModel(AutoformerConfig モデル)
- axk1 — AXK1Model(AXK1Config モデル)
- axk2 — AXK2Model(AXK2Config モデル)
- aya_vision — AyaVisionModel(AyaVisionConfig モデル)
- bailing_hybrid — BailingHybridModel(BailingHybridConfig モデル)
- bamba — BambaModel(BambaConfigモデル)
- bark — BarkModel(BarkConfig モデル)
- bart — BartModel(BartConfig モデル)
- beam — BeamModel(BeamConfig モデル)
- beit — BeitModel(BeitConfig モデル)
- bert — BertModel (BertConfig モデル)
- bert-generation — BertGenerationEncoder (BertGenerationConfig モデル)
- big_bird — BigBirdModel(BigBirdConfig モデル)
- bigbird_pegasus — BigBirdPegasusModel(BigBirdPegasusConfig モデル)
- biogpt — BioGptModel (BioGptConfig モデル)
- bit — BitModel(BitConfig モデル)
- bitnet — BitNetModel(BitNetConfig モデル)
- blenderbot — BlenderbotModel(BlenderbotConfig モデル)
- blenderbot-small — BlenderbotSmallModel (BlenderbotSmallConfig モデル)
- blip — BlipModel(BlipConfig モデル)
- blip-2 — Blip2Model(Blip2Config モデル)
- blip_2_qformer — Blip2QFormerModel(Blip2QFormerConfig モデル)
- bloom — BloomModel(BloomConfig モデル)
- blt — BltModel(BltConfigモデル)
- bridgetower — BridgeTowerModel(BridgeTowerConfig モデル)
- bros — BrosModel(BrosConfig モデル)
- camembert — CamembertModel(CamembertConfig モデル)
- canary — CanaryModel(CanaryConfig モデル)
- canine — CanineModel(CanineConfigモデル)
- chameleon — ChameleonModel(ChameleonConfigモデル)
- chinese_clip — ChineseCLIPModel(ChineseCLIPConfig モデル)
- chinese_clip_vision_model — ChineseCLIPVisionModel(ChineseCLIPVisionConfig モデル)
- clap — ClapModel(ClapConfigモデル)
- clip — CLIPModel (CLIPConfig モデル)
- clip_text_model — CLIPTextModel(CLIPTextConfigモデル)
- clip_vision_model — CLIPVisionModel(CLIPVisionConfig モデル)
- clipseg — CLIPSegModel(CLIPSegConfig モデル)
- clvp — ClvpModelForConditionalGeneration(ClvpConfig モデル)
- codegen — CodeGenModel(CodeGenConfig モデル)
- cohere — CohereModel(CohereConfigモデル)
- cohere2 — Cohere2Model(Cohere2Config モデル)
- cohere2_moe — Cohere2MoeModel (Cohere2MoeConfig モデル)
- cohere2_vision — Cohere2VisionModel(Cohere2VisionConfig モデル)
- cohere_asr — CohereAsrModel(CohereAsrConfig モデル)
- cohere_compass — CohereCompassModel(CohereCompassConfig モデル)
- cohere_compass_text — CohereCompassTextModel (CohereCompassTextConfig モデル)
- cohere_compass_vision — CohereCompassVisionModel (CohereCompassVisionConfig モデル)
- conditional_detr — ConditionalDetrModel(ConditionalDetrConfig モデル)
- convbert — ConvBertModel (ConvBertConfig モデル)
- convnext — ConvNextModel(ConvNextConfig モデル)
- convnextv2 — ConvNextV2Model (ConvNextV2Config モデル)
- cosmos3_edge — Cosmos3EdgeModel(Cosmos3EdgeConfig モデル)
- cosmos3_edge_text — Cosmos3EdgeTextModel (Cosmos3EdgeTextConfig モデル)
- cosmos3_edge_vision — Cosmos3EdgeVisionModel(Cosmos3EdgeVisionConfig モデル)
- cosmos3_omni — Cosmos3OmniModel(Cosmos3OmniConfig モデル)
- cpmant — CpmAntModel(CpmAntConfigモデル)
- csm — CsmForConditionalGeneration(CsmConfigモデル)
- ctrl — CTRLModel(CTRLConfig モデル)
- cvt — CvtModel(CvtConfigモデル)
- cwm — CwmModel(CwmConfigモデル)
- d_fine — DFineModel(DFineConfigモデル)
- dab-detr — DabDetrModel(DabDetrConfig モデル)
- dac — DacModel(DacConfigモデル)
- data2vec-audio — Data2VecAudioModel(Data2VecAudioConfig モデル)
- data2vec-text — Data2VecTextModel(Data2VecTextConfig モデル)
- data2vec-vision — Data2VecVisionModel(Data2VecVisionConfig モデル)
- dbrx — DbrxModel (DbrxConfig モデル)
- deberta — DebertaModel(DebertaConfig モデル)
- deberta-v2 — DebertaV2Model (DebertaV2Config モデル)
- decision_transformer — DecisionTransformerModel(DecisionTransformerConfig モデル)
- deepseek_ocr2 — DeepseekOcr2Model(DeepseekOcr2Config モデル)
- deepseek_v2 — DeepseekV2Model(DeepseekV2Config モデル)
- deepseek_v3 — DeepseekV3Model(DeepseekV3Config モデル)
- deepseek_v32 — DeepseekV32Model(DeepseekV32Config モデル)
- deepseek_v4 — DeepseekV4Model(DeepseekV4Config モデル)
- deepseek_vl — DeepseekVLModel(DeepseekVLConfig モデル)
- deepseek_vl_hybrid — DeepseekVLHybridModel(DeepseekVLHybridConfig モデル)
- deformable_detr — DeformableDetrModel(DeformableDetrConfig モデル)
- deimv2 — Deimv2Model(Deimv2Config モデル)
- deit — DeiTModel(DeiTConfig モデル)
- depth_pro — DepthProModel(DepthProConfig モデル)
- detr — DetrModel(DetrConfig モデル)
- dia — DiaModel(DiaConfig モデル)
- diffllama — DiffLlamaModel(DiffLlamaConfigモデル)
- diffusion_gemma — DiffusionGemmaModel (DiffusionGemmaConfig モデル)
- dinat — DinatModel(DinatConfig モデル)
- dinov2 — Dinov2Model (Dinov2Config モデル)
- dinov2_with_registers — Dinov2WithRegistersModel(Dinov2WithRegistersConfig モデル)
- dinov3_convnext — DINOv3ConvNextModel(DINOv3ConvNextConfig モデル)
- dinov3_vit — DINOv3ViTModel(DINOv3ViTConfig モデル)
- distilbert — DistilBertModel(DistilBertConfig モデル)
- doge — DogeModel(DogeConfig モデル)
- donut-swin — DonutSwinModel(DonutSwinConfig モデル)
- dots1 — Dots1Model(Dots1Config モデル)
- dots3_note — Dots3NoteModel(Dots3NoteConfigモデル)
- dots3_note_text — Dots3NoteTextModel(Dots3NoteTextConfig モデル)
- dpr — DPRQuestionEncoder(DPRConfig モデル)
- dpt — DPTModel(DPTConfig モデル)
- edgetam — EdgeTamModel(EdgeTamConfig モデル)
- edgetam_video — EdgeTamVideoModel(EdgeTamVideoConfig モデル)
- edgetam_vision_model — EdgeTamVisionModel(EdgeTamVisionConfig モデル)
- efficientloftr — EfficientLoFTRModel(EfficientLoFTRConfig モデル)
- efficientnet — EfficientNetModel(EfficientNetConfig モデル)
- electra — ElectraModel(ElectraConfigモデル)
- embedding_gemma2 — EmbeddingGemma2Model(EmbeddingGemma2Config モデル)
- embedding_gemma2_text — EmbeddingGemma2TextModel(EmbeddingGemma2TextConfig モデル)
- emu3 — Emu3Model(Emu3Config モデル)
- encodec — EncodecModel(EncodecConfig モデル)
- ernie — ErnieModel(ErnieConfig モデル)
- ernie4_5 — Ernie4_5Model(Ernie4_5Config モデル)
- ernie4_5_moe — Ernie4_5_MoeModel(Ernie4_5_MoeConfig モデル)
- ernie4_5_vl_moe — Ernie4_5_VLMoeModel(Ernie4_5_VLMoeConfig モデル)
- esm — EsmModel(EsmConfigモデル)
- esmc — EsmcModel(EsmcConfig モデル)
- eurobert — EuroBertModel(EuroBertConfig モデル)
- evolla — EvollaModel(EvollaConfigモデル)
- exaone4 — Exaone4Model(Exaone4Config モデル)
- exaone4_5 — Exaone4_5_Model(Exaone4_5_Config モデル)
- exaone4_5_vision — Exaone4_5_VisionModel(Exaone4_5_VisionConfig モデル)
- exaone_moe — ExaoneMoeModel(ExaoneMoeConfig モデル)
- falcon — FalconModel(FalconConfig モデル)
- falcon_h1 — FalconH1Model (FalconH1Config モデル)
- falcon_mamba — FalconMambaModel (FalconMambaConfig モデル)
- fast_vlm — FastVlmModel(FastVlmConfig モデル)
- fastspeech2_conformer — FastSpeech2ConformerModel(FastSpeech2ConformerConfig モデル)
- fastspeech2_conformer_with_hifigan — FastSpeech2ConformerWithHifiGan(FastSpeech2ConformerWithHifiGanConfig モデル)
- flaubert — FlaubertModel(FlaubertConfig モデル)
- flava — FlavaModel(FlavaConfig モデル)
- flex_olmo — FlexOlmoModel(FlexOlmoConfig モデル)
- flint — FlintModel(FlintConfigモデル)
- florence2 — Florence2Model(Florence2Config モデル)
- fnet — FNetModel (FNetConfig モデル)
- focalnet — FocalNetModel(FocalNetConfig モデル)
- fsmt — FSMTModel(FSMTConfig モデル)
- fun_asr_nano — FunAsrNanoModel(FunAsrNanoConfig モデル)
- fun_asr_nano_encoder — FunAsrNanoEncoder(FunAsrNanoEncoderConfig モデル)
- funnel — FunnelModel または FunnelBaseModel(FunnelConfig モデル)
- fuyu — FuyuModel(FuyuConfig モデル)
- gemma — GemmaModel(GemmaConfig モデル)
- gemma2 — Gemma2Model(Gemma2Config モデル)
- gemma3 — Gemma3Model(Gemma3Config モデル)
- gemma3_text — Gemma3TextModel(Gemma3TextConfig モデル)
- gemma3n — Gemma3nModel(Gemma3nConfig モデル)
- gemma3n_audio —
Gemma3nAudioEncoder(Gemma3nAudioConfig モデル) - gemma3n_text — Gemma3nTextModel(Gemma3nTextConfig モデル)
- gemma3n_vision — TimmWrapperModel(Gemma3nVisionConfig モデル)
- gemma4 — Gemma4Model(Gemma4Config モデル)
- gemma4_audio — Gemma4AudioModel(Gemma4AudioConfig モデル)
- gemma4_text — Gemma4TextModel(Gemma4TextConfig モデル)
- gemma4_unified — Gemma4UnifiedModel(Gemma4UnifiedConfig モデル)
- gemma4_unified_text — Gemma4UnifiedTextModel(Gemma4UnifiedTextConfig モデル)
- gemma4_vision — Gemma4VisionModel(Gemma4VisionConfig モデル)
- git — GitModel(GitConfig モデル)
- glm — GlmModel(GlmConfig モデル)
- glm4 — Glm4Model(Glm4Config モデル)
- glm46v — Glm46VModel(Glm46VConfig モデル)
- glm4_moe — Glm4MoeModel(Glm4MoeConfig モデル)
- glm4_moe_lite — Glm4MoeLiteModel(Glm4MoeLiteConfig モデル)
- glm4v — Glm4vModel(Glm4vConfig モデル)
- glm4v_moe — Glm4vMoeModel(Glm4vMoeConfig モデル)
- glm4v_moe_text — Glm4vMoeTextModel(Glm4vMoeTextConfig モデル)
- glm4v_moe_vision — Glm4vMoeVisionModel(Glm4vMoeVisionConfig モデル)
- glm4v_text — Glm4vTextModel(Glm4vTextConfig モデル)
- glm4v_vision — Glm4vVisionModel(Glm4vVisionConfig モデル)
- glm5_next — Glm5NextModel(Glm5NextConfig モデル)
- glm5_next_text — Glm5NextTextModel(Glm5NextTextConfig モデル)
- glm5_next_vision — Glm5NextVisionModel(Glm5NextVisionConfig モデル)
- glm_image — GlmImageModel(GlmImageConfig モデル)
- glm_image_text — GlmImageTextModel(GlmImageTextConfig モデル)
- glm_image_vision — GlmImageVisionModel(GlmImageVisionConfig モデル)
- glm_image_vqmodel — GlmImageVQVAE(GlmImageVQVAEConfig モデル)
- glm_moe_dsa — GlmMoeDsaModel(GlmMoeDsaConfig モデル)
- glm_ocr — GlmOcrModel(GlmOcrConfig モデル)
- glm_ocr_text — GlmOcrTextModel(GlmOcrTextConfig モデル)
- glm_ocr_vision — GlmOcrVisionModel(GlmOcrVisionConfig モデル)
- glmasr — GlmAsrModel(GlmAsrConfig モデル)
- glmasr_encoder — GlmAsrEncoder(GlmAsrEncoderConfig モデル)
- glmga — Glm46VModel(GlmgaConfig モデル)
- glpn — GLPNModel(GLPNConfigモデル)
- got_ocr2 — GotOcr2Model(GotOcr2Config モデル)
- gpt-sw3 — GPT2Model(GPT2Configモデル)
- gpt2 — GPT2Model (GPT2Config モデル)
- gpt_bigcode — GPTBigCodeModel(GPTBigCodeConfig モデル)
- gpt_neo — GPTNeoModel(GPTNeoConfigモデル)
- gpt_neox — GPTNeoXModel(GPTNeoXConfig モデル)
- gpt_neox_japanese — GPTNeoXJapaneseModel(GPTNeoXJapaneseConfig モデル)
- gpt_oss — GptOssModel(GptOssConfig モデル)
- gptj — GPTJModel (GPTJConfig モデル)
- gradium_phonon_tts — GradiumPhononTTSModel(GradiumPhononTTSConfig モデル)
- granite — GraniteModel(GraniteConfigモデル)
- granite4_vision — Granite4VisionModel(Granite4VisionConfig モデル)
- granite_docling — GraniteDoclingModel(GraniteDoclingConfig モデル)
- granite_docling_text — GraniteDoclingTextModel(GraniteDoclingTextConfig モデル)
- granite_docling_vision — GraniteDoclingVisionModel(GraniteDoclingVisionConfig モデル)
- granite_speech — GraniteSpeechModel(GraniteSpeechConfig モデル)
- granite_speech5_ctc — GraniteSpeech5ForCTC(GraniteSpeech5CTCConfig モデル)
- granite_speech5_encoder — GraniteSpeech5Encoder(GraniteSpeech5EncoderConfig モデル)
- granite_speech_plus — GraniteSpeechPlusModel(GraniteSpeechPlusConfig モデル)
- granite_swa — GraniteSWAModel (GraniteSWAConfig モデル)
- granitemoe — GraniteMoeModel(GraniteMoeConfigモデル)
- granitemoe_swa — GraniteMoeSWAModel (GraniteMoeSWAConfig モデル)
- granitemoehybrid — GraniteMoeHybridModel(GraniteMoeHybridConfig モデル)
- granitemoeshared — GraniteMoeSharedModel(GraniteMoeSharedConfig モデル)
- grounding-dino — GroundingDinoModel(GroundingDinoConfig モデル)
- groupvit — GroupViTModel(GroupViTConfig モデル)
- gte — GteModel(GteConfigモデル)
- helium — HeliumModel(HeliumConfigモデル)
- hgnet_v2 — HGNetV2Backbone(HGNetV2Config モデル)
- hiera — HieraModel(HieraConfigモデル)
- higgs_audio_v2 — HiggsAudioV2ForConditionalGeneration(HiggsAudioV2Config モデル)
- higgs_audio_v2_tokenizer — HiggsAudioV2TokenizerModel(HiggsAudioV2TokenizerConfig モデル)
- hrm_text — HrmTextModel(HrmTextConfig モデル)
- hubert — HubertModel(HubertConfig モデル)
- hunyuan_v1_dense — HunYuanDenseV1Model(HunYuanDenseV1Config モデル)
- hunyuan_v1_moe — HunYuanMoEV1Model(HunYuanMoEV1Config モデル)
- hunyuan_vl — HunYuanVLModel(HunYuanVLConfig モデル)
- hunyuan_vl_text — HunYuanVLTextModel (HunYuanVLTextConfig モデル)
- hunyuan_vl_vision — HunYuanVLVisionTransformer(HunYuanVLVisionConfig モデル)
- hy_v3 — HYV3Model(HYV3Config モデル)
- hy_v4 — HYV4Model(HYV4Config モデル)
- hyperclovax — HyperCLOVAXModel(HyperCLOVAXConfig モデル)
- hyperclovax_vision_v2 — HyperCLOVAXVisionV2Model(HyperCLOVAXVisionV2Config モデル)
- ibert — IBertModel(IBertConfigモデル)
- idefics — IdeficsModel(IdeficsConfig モデル)
- idefics2 — Idefics2Model(Idefics2Config モデル)
- idefics3 — Idefics3Model(Idefics3Config モデル)
- idefics3_vision — Idefics3VisionTransformer(Idefics3VisionConfig モデル)
- ijepa — IJepaModel(IJepaConfig モデル)
- imagegpt — ImageGPTModel(ImageGPTConfig モデル)
- informer — InformerModel(InformerConfig モデル)
- inkling2 — Inkling2Model(Inkling2Configモデル)
- inkling2_text — Inkling2TextModel(Inkling2TextConfig モデル)
- inkling2_vision — Inkling2VisionModel(Inkling2VisionConfig モデル)
- inkling_mm_model — InklingModel(InklingConfig モデル)
- inkling_text — InklingTextModel(InklingTextConfig モデル)
- instructblip — InstructBlipModel(InstructBlipConfig モデル)
- instructblipvideo — InstructBlipVideoModel(InstructBlipVideoConfig モデル)
- internlm3 — LlamaModel(InternLM3Config モデル)
- internvl — InternVLModel(InternVLConfig モデル)
- internvl_vision — InternVLVisionModel(InternVLVisionConfig モデル)
- iquestcoder — LlamaModel(IQuestCoderConfigモデル)
- jais2 — Jais2Model(Jais2Config モデル)
- jamba — JambaModel(JambaConfig モデル)
- janus — JanusModel(JanusConfig モデル)
- jetmoe — JetMoeModel(JetMoeConfigモデル)
- jina_embeddings_v3 — JinaEmbeddingsV3Model(JinaEmbeddingsV3Configモデル)
- kimi_k25 — Kimi_K25Model(Kimi_K25Config モデル)
- kimi_k25_vision — Kimi_K25VisionModel(Kimi_K25VisionConfig モデル)
- kimi_linear — KimiLinearModel(KimiLinearConfig モデル)
- kolibri1 — Kolibri1Model(Kolibri1Config モデル)
- kosmos-2 — Kosmos2Model(Kosmos2Config モデル)
- kosmos-2.5 — Kosmos2_5Model(Kosmos2_5Config モデル)
- kyutai_speech_to_text — KyutaiSpeechToTextModel (KyutaiSpeechToTextConfig モデル)
- laguna — LagunaModel(LagunaConfig モデル)
- lasr_ctc — LasrForCTC(LasrCTCConfig モデル)
- lasr_encoder — LasrEncoder(LasrEncoderConfig モデル)
- layoutlm — LayoutLMModel(LayoutLMConfig モデル)
- layoutlmv2 — LayoutLMv2Model(LayoutLMv2Config モデル)
- layoutlmv3 — LayoutLMv3Model(LayoutLMv3Config モデル)
- led — LEDModel (LEDConfig モデル)
- levit — LevitModel(LevitConfigモデル)
- lfm2 — Lfm2Model(Lfm2Config モデル)
- lfm2_moe — Lfm2MoeModel(Lfm2MoeConfig モデル)
- lfm2_vl — Lfm2VlModel(Lfm2VlConfig モデル)
- lfm3 — Lfm3Model(Lfm3Configモデル)
- lfm3_vl — Lfm3VlModel (Lfm3VlConfig モデル)
- lfm3_vl_vision — Lfm3VlVisionModel(Lfm3VlVisionConfig モデル)
- lightglue — LightGlueForKeypointMatching(LightGlueConfig モデル)
- lighton_ocr — LightOnOcrModel(LightOnOcrConfig モデル)
- lilt — LiltModel(LiltConfigモデル)
- llama — LlamaModel(LlamaConfigモデル)
- llama4 — Llama4ForConditionalGeneration(Llama4Config モデル)
- llama4_text — Llama4TextModel(Llama4TextConfig モデル)
- llava — LlavaModel (LlavaConfig モデル)
- llava_next — LlavaNextModel(LlavaNextConfig モデル)
- llava_next_video — LlavaNextVideoModel(LlavaNextVideoConfig モデル)
- llava_onevision — LlavaOnevisionModel(LlavaOnevisionConfig モデル)
- longcat_flash — LongcatFlashModel(LongcatFlashConfig モデル)
- longformer — LongformerModel(LongformerConfig モデル)
- longt5 — LongT5Model(LongT5Config モデル)
- luke — LukeModel(LukeConfig モデル)
- lw_detr — LwDetrModel(LwDetrConfig モデル)
- lxmert — LxmertModel(LxmertConfigモデル)
- m2m_100 — M2M100Model(M2M100Config モデル)
- mamba — MambaModel(MambaConfigモデル)
- mamba2 — Mamba2Model(Mamba2Config モデル)
- marian — MarianModel(MarianConfigモデル)
- markuplm — MarkupLMModel(MarkupLMConfig モデル)
- mask2former — Mask2FormerModel(Mask2FormerConfig モデル)
- maskformer — MaskFormerModel(MaskFormerConfig モデル)
- maskformer-swin —
MaskFormerSwinModel(MaskFormerSwinConfig モデル) - mbart — MBartModel (MBartConfig モデル)
- megatron-bert — MegatronBertModel (MegatronBertConfig モデル)
- mellum — MellumModel(MellumConfig モデル)
- metaclip_2 — MetaClip2Model(MetaClip2Config モデル)
- mgp-str — MgpstrForSceneTextRecognition(MgpstrConfigモデル)
- mimi — MimiModel(MimiConfig モデル)
- mimi_decoder — MimiDecoderModel(MimiDecoderConfig モデル)
- mimi_encoder — MimiEncoderModel(MimiEncoderConfig モデル)
- mimo_v2_flash — MiMoV2FlashModel(MiMoV2FlashConfig モデル)
- minicpm3 — MiniCPM3Model(MiniCPM3Config モデル)
- minicpm_ocr — MiniCPMOcrModel(MiniCPMOcrConfig モデル)
- minicpm_ocr_vision — MiniCPMOcrVisionModel(MiniCPMOcrVisionConfig モデル)
- minicpmv4_6 — MiniCPMV4_6Model(MiniCPMV4_6Config モデル)
- minicpmv4_7 — MiniCPMV4_7Model(MiniCPMV4_7Config モデル)
- minicpmv4_7_vision — MiniCPMV4_7VisionModel(MiniCPMV4_7VisionConfig モデル)
- minimax — MiniMaxModel(MiniMaxConfig モデル)
- minimax_m2 — MiniMaxM2Model(MiniMaxM2Config モデル)
- minimax_m3_vl — MiniMaxM3VLModel(MiniMaxM3VLConfig モデル)
- minimax_m3_vl_text — MiniMaxM3VLTextModel(MiniMaxM3VLTextConfig モデル)
- ministral — MinistralModel(MinistralConfig モデル)
- ministral3 — Ministral3Model(Ministral3Config モデル)
- mistral — MistralModel(MistralConfigモデル)
- mistral3 — Mistral3Model(Mistral3Config モデル)
- mistral4 — Mistral4Model(Mistral4Config モデル)
- mistral_large4 — MistralLarge4Model(MistralLarge4Config モデル)
- mistral_large4_text — MistralLarge4TextModel(MistralLarge4TextConfig モデル)
- mixtral — MixtralModel(MixtralConfig モデル)
- mlcd — MLCDVisionModel(MLCDVisionConfig モデル)
- mlcd_vision_model — MLCDVisionModel(MLCDVisionConfig モデル)
- mllama — MllamaModel(MllamaConfig モデル)
- mm-grounding-dino — MMGroundingDinoModel(MMGroundingDinoConfig モデル)
- mobilebert — MobileBertModel(MobileBertConfig モデル)
- mobilenet_v1 — MobileNetV1Model (MobileNetV1Config モデル)
- mobilenet_v2 — MobileNetV2Model (MobileNetV2Config モデル)
- mobilevit — MobileViTModel(MobileViTConfig モデル)
- mobilevitv2 — MobileViTV2Model(MobileViTV2Config モデル)
- modernbert — ModernBertModel(ModernBertConfig モデル)
- modernbert-decoder — ModernBertDecoderModel(ModernBertDecoderConfig モデル)
- modernvbert — ModernVBertModel(ModernVBertConfig モデル)
- molmo2 — Molmo2Model(Molmo2Config モデル)
- molmo2_text — Molmo2TextModel(Molmo2TextConfig モデル)
- moonshine — MoonshineModel(MoonshineConfig モデル)
- moonshine_streaming — MoonshineStreamingModel(MoonshineStreamingConfig モデル)
- moshi — MoshiModel(MoshiConfigモデル)
- moss_transcribe_diarize — MossTranscribeDiarizeModel(MossTranscribeDiarizeConfig モデル)
- moss_transcribe_diarize_audio — MossTranscribeDiarizeAudioModel (MossTranscribeDiarizeAudioConfig モデル)
- mpnet — MPNetModel(MPNetConfig モデル)
- mpt — MptModel(MptConfigモデル)
- mra — MraModel(MraConfigモデル)
- mt5 — MT5Model(MT5Config モデル)
- muse_glimmer — MuseGlimmerModel(MuseGlimmerConfig モデル)
- muse_glimmer_assistant — MuseGlimmerAssistantModel(MuseGlimmerAssistantConfig モデル)
- muse_glimmer_text — MuseGlimmerTextModel(MuseGlimmerTextConfig モデル)
- muse_glimmer_vision — MuseGlimmerVisionModel(MuseGlimmerVisionConfig モデル)
- muse_spark — MuseSparkModel(MuseSparkConfigモデル)
- muse_spark_audio — MuseSparkAudioModel(MuseSparkAudioConfig モデル)
- muse_spark_text — MuseSparkTextModel(MuseSparkTextConfig モデル)
- muse_spark_vision — MuseSparkVisionModel(MuseSparkVisionConfig モデル)
- musicflamingo — MusicFlamingoModel(MusicFlamingoConfig モデル)
- musicgen — MusicgenModel(MusicgenConfig モデル)
- musicgen_melody — MusicgenMelodyModel(MusicgenMelodyConfig モデル)
- mvp — MvpModel(MvpConfigモデル)
- nanochat — NanoChatModel(NanoChatConfig モデル)
- nemo_speechlm — Nemotron3_5TranscribeModel(Nemotron3_5TranscribeConfig モデル)
- nemotron — NemotronModel(NemotronConfig モデル)
- nemotron3_5_asr — Nemotron3_5AsrForRNNT(Nemotron3_5AsrConfig モデル)
- nemotron3_5_transcribe_asr_audio — Nemotron3_5TranscribeAsrAudioModel(Nemotron3_5TranscribeAsrAudioConfig モデル)
- nemotron3_5_transcribe_audio — Nemotron3_5TranscribeAudioModel(Nemotron3_5TranscribeAudioConfig モデル)
- nemotron3_5_transcribe_word_timestamp — Nemotron3_5TranscribeWordTimestampModel(Nemotron3_5TranscribeWordTimestampConfig モデル)
- nemotron3_diarization — Nemotron3DiarizationModel(Nemotron3DiarizationConfig モデル)
- nemotron3_diarization_audio — Nemotron3DiarizationAudioModel(Nemotron3DiarizationAudioConfig モデル)
- nemotron_asr_streaming — NemotronAsrStreamingForRNNT(NemotronAsrStreamingConfig モデル)
- nemotron_asr_streaming_encoder — NemotronAsrStreamingEncoder (NemotronAsrStreamingEncoderConfig モデル)
- nemotron_h — NemotronHModel(NemotronHConfig モデル)
- neomme — NeoMMEModel(NeoMMEConfigモデル)
- neucodec — NeuCodecModel(NeuCodecConfig モデル)
- nllb-moe — NllbMoeModel(NllbMoeConfig モデル)
- nomic_bert — NomicBertModel(NomicBertConfig モデル)
- nystromformer — NystromformerModel(NystromformerConfig モデル)
- olmo — OlmoModel(OlmoConfig モデル)
- olmo2 — Olmo2Model(Olmo2Config モデル)
- olmo3 — Olmo3Model(Olmo3Config モデル)
- olmo_hybrid — OlmoHybridModel(OlmoHybridConfig モデル)
- olmoe — OlmoeModel(OlmoeConfig モデル)
- omdet-turbo — OmDetTurboForObjectDetection(OmDetTurboConfig モデル)
- oneformer — OneFormerModel(OneFormerConfig モデル)
- openai-gpt — OpenAIGPTModel(OpenAIGPTConfig モデル)
- openai_privacy_filter — OpenAIPrivacyFilterModel(OpenAIPrivacyFilterConfig モデル)
- opt — OPTModel(OPTConfig モデル)
- ovis2 — Ovis2Model(Ovis2Config モデル)
- owlv2 — Owlv2Model (Owlv2Config モデル)
- owlvit — OwlViTModel(OwlViTConfig モデル)
- paddleocr_vl — PaddleOCRVLModel(PaddleOCRVLConfig モデル)
- paligemma — PaliGemmaModel (PaliGemmaConfig モデル)
- parakeet_ctc — ParakeetForCTC(ParakeetCTCConfig モデル)
- parakeet_encoder — ParakeetEncoder(ParakeetEncoderConfigモデル)
- parakeet_rnnt — ParakeetForRNNT(ParakeetRNNTConfigモデル)
- parakeet_tdt — ParakeetForTDT(ParakeetTDTConfig モデル)
- patchtsmixer — PatchTSMixerModel(PatchTSMixerConfig モデル)
- patchtst — PatchTSTModel(PatchTSTConfig モデル)
- pe_audio — PeAudioModel(PeAudioConfig モデル)
- pe_audio_encoder — PeAudioEncoder(PeAudioEncoderConfig モデル)
- pe_audio_video — PeAudioVideoModel(PeAudioVideoConfig モデル)
- pe_audio_video_encoder — PeAudioVideoEncoder(PeAudioVideoEncoderConfig モデル)
- pe_video — PeVideoModel(PeVideoConfig モデル)
- pe_video_encoder — PeVideoEncoder(PeVideoEncoderConfig モデル)
- pegasus — PegasusModel(PegasusConfig モデル)
- pegasus_x — PegasusXModel(PegasusXConfig モデル)
- perceiver — PerceiverModel(PerceiverConfigモデル)
- perception_lm — PerceptionLMModel(PerceptionLMConfig モデル)
- persimmon — PersimmonModel(PersimmonConfig モデル)
- phi — PhiModel(PhiConfig モデル)
- phi3 — Phi3Model(Phi3Config モデル)
- phi4_multimodal — Phi4MultimodalModel(Phi4MultimodalConfig モデル)
- phimoe — PhimoeModel(PhimoeConfig モデル)
- pi0 — PI0Model(PI0Configモデル)
- pixio — PixioModel(PixioConfig モデル)
- pixtral — PixtralVisionModel(PixtralVisionConfig モデル)
- plbart — PLBartModel(PLBartConfig モデル)
- pocket_tts — PocketTTSModel(PocketTTSConfig モデル)
- pocket_tts_audio_tokenizer — PocketTTSAudioTokenizerModel(PocketTTSAudioTokenizerConfig モデル)
- pocket_tts_audio_tokenizer_decoder — PocketTTSAudioTokenizerDecoderModel(PocketTTSAudioTokenizerDecoderConfig モデル)
- pocket_tts_audio_tokenizer_encoder — PocketTTSAudioTokenizerEncoderModel(PocketTTSAudioTokenizerEncoderConfig モデル)
- poolformer — PoolFormerModel(PoolFormerConfig モデル)
- pp_doclayout_v3 — PPDocLayoutV3Model(PPDocLayoutV3Config モデル)
- pp_doclayout_v4 — PPDocLayoutV4Model(PPDocLayoutV4Config モデル)
- pp_ocrv5_mobile_rec — PPOCRV5MobileRecModel(PPOCRV5MobileRecConfig モデル)
- pp_ocrv5_server_rec — PPOCRV5ServerRecModel(PPOCRV5ServerRecConfig モデル)
- pp_ocrv6_medium_det — PPOCRV6MediumDetModel(PPOCRV6MediumDetConfig モデル)
- pp_ocrv6_small_det — PPOCRV6SmallDetModel(PPOCRV6SmallDetConfig モデル)
- pp_ocrv6_small_rec — PPOCRV6SmallRecModel(PPOCRV6SmallRecConfig モデル)
- pp_ocrv6_tiny_rec — PPOCRV6TinyRecModel(PPOCRV6TinyRecConfig モデル)
- prophetnet — ProphetNetModel(ProphetNetConfig モデル)
- pvt — PvtModel (PvtConfig モデル)
- pvt_v2 — PvtV2Model(PvtV2Config モデル)
- qianfan_ocr — QianfanOCRModel(QianfanOCRConfig モデル)
- qianfan_ocr_vision — QianfanOCRVisionModel(QianfanOCRVisionConfig モデル)
- qwen2 — Qwen2Model(Qwen2Configモデル)
- qwen2_5_omni_thinker — Qwen2_5OmniThinkerModel(Qwen2_5OmniThinkerConfig モデル)
- qwen2_5_vl — Qwen2_5_VLModel(Qwen2_5_VLConfig モデル)
- qwen2_5_vl_text — Qwen2_5_VLTextModel(Qwen2_5_VLTextConfig モデル)
- qwen2_5_vl_vision — Qwen2_5_VisionTransformerPretrainedModel(Qwen2_5_VLVisionConfig モデル)
- qwen2_audio — Qwen2AudioModel(Qwen2AudioConfig モデル)
- qwen2_audio_encoder — Qwen2AudioEncoder(Qwen2AudioEncoderConfig モデル)
- qwen2_moe — Qwen2MoeModel(Qwen2MoeConfig モデル)
- qwen2_vl — Qwen2VLModel(Qwen2VLConfig モデル)
- qwen2_vl_text — Qwen2VLTextModel(Qwen2VLTextConfig モデル)
- qwen3 — Qwen3Model(Qwen3Config モデル)
- qwen3_5 — Qwen3_5Model(Qwen3_5Config モデル)
- qwen3_5_moe — Qwen3_5MoeModel(Qwen3_5MoeConfig モデル)
- qwen3_5_moe_text — Qwen3_5MoeTextModel(Qwen3_5MoeTextConfig モデル)
- qwen3_5_moe_vision — Qwen3_5MoeVisionModel(Qwen3_5MoeVisionConfig モデル)
- qwen3_5_text — Qwen3_5TextModel(Qwen3_5TextConfig モデル)
- qwen3_5_vision — Qwen3_5VisionModel(Qwen3_5VisionConfig モデル)
- qwen3_asr — Qwen3ASRModel(Qwen3ASRConfig モデル)
- qwen3_asr_encoder — Qwen3ASREncoder(Qwen3ASREncoderConfig モデル)
- qwen3_moe — Qwen3MoeModel(Qwen3MoeConfig モデル)
- qwen3_next — Qwen3NextModel(Qwen3NextConfig モデル)
- qwen3_omni_moe_thinker — Qwen3OmniMoeThinkerModel(Qwen3OmniMoeThinkerConfig モデル)
- qwen3_vl — Qwen3VLModel(Qwen3VLConfig モデル)
- qwen3_vl_moe — Qwen3VLMoeModel(Qwen3VLMoeConfig モデル)
- qwen3_vl_moe_text — Qwen3VLMoeTextModel (Qwen3VLMoeTextConfig モデル)
- qwen3_vl_moe_vision — Qwen3VLMoeVisionModel(Qwen3VLMoeVisionConfig モデル)
- qwen3_vl_text — Qwen3VLTextModel(Qwen3VLTextConfig モデル)
- qwen3_vl_vision — Qwen3VLVisionModel(Qwen3VLVisionConfig モデル)
- qwen4_exp — Qwen4ExpModel(Qwen4ExpConfig モデル)
- qwen4_exp_text — Qwen4ExpTextModel(Qwen4ExpTextConfig モデル)
- qwen4_exp_vision — Qwen4ExpVisionModel(Qwen4ExpVisionConfig モデル)
- radio — RadioModel(RadioConfigモデル)
- recurrent_gemma — RecurrentGemmaModel(RecurrentGemmaConfig モデル)
- reformer — ReformerModel(ReformerConfigモデル)
- regnet — RegNetModel(RegNetConfig モデル)
- rembert — RemBertModel (RemBertConfig モデル)
- resnet — ResNetModel(ResNetConfig モデル)
- rf_detr — RfDetrModel(RfDetrConfig モデル)
- roberta — RobertaModel(RobertaConfig モデル)
- roberta-prelayernorm — RobertaPreLayerNormModel(RobertaPreLayerNormConfig モデル)
- roc_bert — RoCBertModel(RoCBertConfig モデル)
- roformer — RoFormerModel(RoFormerConfig モデル)
- rt_detr — RTDetrModel(RTDetrConfigモデル)
- rt_detr_v2 — RTDetrV2Model(RTDetrV2Config モデル)
- rwkv — RwkvModel(RwkvConfigモデル)
- sam — SamModel(SamConfig モデル)
- sam2 — Sam2Model(Sam2Config モデル)
- sam2_hiera_det_model — Sam2HieraDetModel(Sam2HieraDetConfig モデル)
- sam2_video — Sam2VideoModel (Sam2VideoConfig モデル)
- sam2_vision_model — Sam2VisionModel(Sam2VisionConfig モデル)
- sam3 — Sam3Model(Sam3Config モデル)
- sam3_lite_text — Sam3LiteTextModel(Sam3LiteTextConfig モデル)
- sam3_lite_text_text_model — Sam3LiteTextTextModel(Sam3LiteTextTextConfig モデル)
- sam3_tracker — Sam3TrackerModel(Sam3TrackerConfig モデル)
- sam3_tracker_video — Sam3TrackerVideoModel(Sam3TrackerVideoConfig モデル)
- sam3_video — Sam3VideoModel(Sam3VideoConfig モデル)
- sam3_vision_model — Sam3VisionModel(Sam3VisionConfigモデル)
- sam3_vit_model — Sam3ViTModel(Sam3ViTConfig モデル)
- sam_hq — SamHQModel(SamHQConfig モデル)
- sam_hq_vision_model — SamHQVisionModel(SamHQVisionConfig モデル)
- sam_vision_model — SamVisionModel(SamVisionConfig モデル)
- sapiens2 — Sapiens2Model(Sapiens2Config モデル)
- seamless_m4t — SeamlessM4TModel(SeamlessM4TConfig モデル)
- seamless_m4t_v2 — SeamlessM4Tv2Model(SeamlessM4Tv2Config モデル)
- seed_oss — SeedOssModel(SeedOssConfig モデル)
- segformer — SegformerModel(SegformerConfig モデル)
- seggpt — SegGptModel(SegGptConfigモデル)
- sew — SEWModel(SEWConfigモデル)
- sew-d — SEWDModel(SEWDConfigモデル)
- siglip — SiglipModel(SiglipConfig モデル)
- siglip2 — Siglip2Model(Siglip2Config モデル)
- siglip2_vision_model — Siglip2VisionModel(Siglip2VisionConfig モデル)
- siglip_vision_model — SiglipVisionModel(SiglipVisionConfig モデル)
- smollm3 — SmolLM3Model(SmolLM3Config モデル)
- smolvlm — SmolVLMModel(SmolVLMConfig モデル)
- smolvlm_vision — SmolVLMVisionTransformer(SmolVLMVisionConfig モデル)
- solar_open — SolarOpenModel(SolarOpenConfig モデル)
- speech_to_text — Speech2TextModel(Speech2TextConfig モデル)
- speecht5 — SpeechT5Model(SpeechT5Config モデル)
- splinter — SplinterModel(SplinterConfigモデル)
- squeezebert — SqueezeBertModel (SqueezeBertConfig モデル)
- stablelm — StableLmModel (StableLmConfig モデル)
- starcoder2 — Starcoder2Model(Starcoder2Config モデル)
- step3p7 — Step3p7Model(Step3p7Config モデル)
- swiftformer — SwiftFormerModel(SwiftFormerConfig モデル)
- swin — SwinModel (SwinConfig モデル)
- swin2sr — Swin2SRModel(Swin2SRConfigモデル)
- swinv2 — Swinv2Model(Swinv2Config モデル)
- switch_transformers — SwitchTransformersModel(SwitchTransformersConfig モデル)
- t5 — T5Model (T5Config モデル)
- t5gemma — T5GemmaModel (T5GemmaConfig モデル)
- t5gemma2 — T5Gemma2Model (T5Gemma2Config モデル)
- t5gemma2_encoder —
T5Gemma2Encoder(T5Gemma2EncoderConfig モデル) - table-transformer — TableTransformerModel(TableTransformerConfig モデル)
- tapas — TapasModel(TapasConfig モデル)
- textnet — TextNetModel(TextNetConfig モデル)
- time_series_transformer — TimeSeriesTransformerModel(TimeSeriesTransformerConfig モデル)
- timesfm — TimesFmModel(TimesFmConfig モデル)
- timesfm2_5 — TimesFm2_5Model(TimesFm2_5Config モデル)
- timesformer — TimesformerModel(TimesformerConfig モデル)
- timm_backbone — TimmBackbone(TimmBackboneConfig モデル)
- timm_wrapper — TimmWrapperModel(TimmWrapperConfig モデル)
- tipsv2 — Tipsv2Model(Tipsv2Config モデル)
- tipsv2_text_model — Tipsv2TextModel(Tipsv2TextConfig モデル)
- tipsv2_vision_model — Tipsv2VisionModel(Tipsv2VisionConfig モデル)
- tvp — TvpModel(TvpConfigモデル)
- udop — UdopModel (UdopConfig モデル)
- ultrabert — UltrabertModel(UltrabertConfig モデル)
- ultravbert — UltraVBertModel(UltraVBertConfig モデル)
- umt5 — UMT5Model(UMT5Config モデル)
- unispeech — UniSpeechModel(UniSpeechConfig モデル)
- unispeech-sat — UniSpeechSatModel(UniSpeechSatConfig モデル)
- univnet — UnivNetModel(UnivNetConfig モデル)
- unlimited_ocr — UnlimitedOcrModel(UnlimitedOcrConfig モデル)
- uvdoc — UVDocModel(UVDocConfig モデル)
- vaultgemma — VaultGemmaModel(VaultGemmaConfig モデル)
- vibevoice — VibeVoiceForConditionalGeneration(VibeVoiceConfig モデル)
- vibevoice_acoustic_tokenizer — VibeVoiceAcousticTokenizerModel(VibeVoiceAcousticTokenizerConfig モデル)
- vibevoice_acoustic_tokenizer_decoder — VibeVoiceAcousticTokenizerDecoderModel(VibeVoiceAcousticTokenizerDecoderConfig モデル)
- vibevoice_acoustic_tokenizer_encoder — VibeVoiceAcousticTokenizerEncoderModel(VibeVoiceAcousticTokenizerEncoderConfig モデル)
- vibevoice_asr — VibeVoiceAsrModel(VibeVoiceAsrConfig モデル)
- video_llama_3 — VideoLlama3Model(VideoLlama3Config モデル)
- video_llama_3_vision — VideoLlama3VisionModel(VideoLlama3VisionConfig モデル)
- video_llava — VideoLlavaModel(VideoLlavaConfig モデル)
- videomae — VideoMAEModel(VideoMAEConfig モデル)
- videoprism — VideoPrismClipModel(VideoPrismConfigモデル)
- videoprism_vision_model — VideoPrismVisionModel(VideoPrismVisionConfig モデル)
- vilt — ViltModel(ViltConfig モデル)
- vipllava — VipLlavaModel(VipLlavaConfig モデル)
- vision-text-dual-encoder — VisionTextDualEncoderModel(VisionTextDualEncoderConfig モデル)
- visual_bert — VisualBertModel(VisualBertConfig モデル)
- vit — ViTModel(ViTConfig モデル)
- vit_mae — ViTMAEModel (ViTMAEConfig モデル)
- vit_msn — ViTMSNModel(ViTMSNConfig モデル)
- vitdet — VitDetModel(VitDetConfig モデル)
- vits — VitsModel(VitsConfig モデル)
- vivit — VivitModel(VivitConfigモデル)
- vjepa2 — VJEPA2Model(VJEPA2Config モデル)
- voxtral — VoxtralModel(VoxtralConfig モデル)
- voxtral_encoder — VoxtralEncoder(VoxtralEncoderConfig モデル)
- voxtral_realtime — VoxtralRealtimeModel(VoxtralRealtimeConfig モデル)
- voxtral_realtime_encoder — VoxtralRealtimeEncoder(VoxtralRealtimeEncoderConfig モデル)
- voxtral_realtime_text —
VoxtralRealtimeTextModel(VoxtralRealtimeTextConfig モデル) - wav2vec2 — Wav2Vec2Model(Wav2Vec2Config モデル)
- wav2vec2-bert — Wav2Vec2BertModel (Wav2Vec2BertConfig モデル)
- wav2vec2-conformer — Wav2Vec2ConformerModel(Wav2Vec2ConformerConfig モデル)
- wavlm — WavLMModel(WavLMConfig モデル)
- whisper — WhisperModel(WhisperConfigモデル)
- xclip — XCLIPModel(XCLIPConfigモデル)
- xcodec — XcodecModel(XcodecConfig モデル)
- xcodec2 — Xcodec2Model(Xcodec2Configモデル)
- xglm — XGLMModel(XGLMConfig モデル)
- xlm — XLMModel(XLMConfigモデル)
- xlm-roberta — XLMRobertaModel(XLMRobertaConfig モデル)
- xlm-roberta-xl — XLMRobertaXLModel(XLMRobertaXLConfig モデル)
- xlnet — XLNetModel(XLNetConfigモデル)
- xlstm — xLSTMModel(xLSTMConfig モデル)
- xmod — XmodModel(XmodConfigモデル)
- yolos — YolosModel(YolosConfig モデル)
- yoso — YosoModel(YosoConfigモデル)
- youtu — YoutuModel(YoutuConfig モデル)
- zamba — ZambaModel(ZambaConfigモデル)
- zamba2 — Zamba2Model (Zamba2Config モデル)
- zaya — ZayaModel(ZayaConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModel
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModel.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModel.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True汎用事前学習クラス
以下は、事前学習ヘッド付きのモデルをインスタンス化するために利用可能なオートクラスです。
AutoModelForPreTraining
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスの一つ(事前学習ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- AlbertConfig 設定クラス: AlbertForPreTraining (AlbertConfig モデル)
- AudioFlamingo3Config 設定クラス: AudioFlamingo3ForConditionalGeneration (AudioFlamingo3Config モデル)
- BartConfig 設定クラス: BartForConditionalGeneration (BartConfig モデル)
- BertConfig 設定クラス: BertForPreTraining (BertConfig モデル)
- BigBirdConfig 設定クラス: BigBirdForPreTraining (BigBirdConfig モデル)
- BloomConfig 設定クラス: BloomForCausalLM (BloomConfig モデル)
- CTRLConfig 設定クラス: CTRLLMHeadModel (CTRLConfig モデル)
- CamembertConfig 設定クラス: CamembertForMaskedLM (CamembertConfig モデル)
- ColModernVBertConfig 設定クラス: ColModernVBertForRetrieval (ColModernVBertConfig モデル)
- ColPaliConfig 設定クラス: ColPaliForRetrieval (ColPaliConfig モデル)
- ColQwen2Config 設定クラス: ColQwen2ForRetrieval (ColQwen2Config モデル)
- Data2VecTextConfig 設定クラス: Data2VecTextForMaskedLM (Data2VecTextConfig モデル)
- DebertaConfig 設定クラス: DebertaForMaskedLM (DebertaConfig モデル)
- DebertaV2Config 設定クラス: DebertaV2ForMaskedLM (DebertaV2Config モデル)
- DistilBertConfig 設定クラス: DistilBertForMaskedLM (DistilBertConfig モデル)
- ElectraConfig 設定クラス: ElectraForPreTraining (ElectraConfig モデル)
- ErnieConfig 設定クラス: ErnieForPreTraining (ErnieConfig モデル)
- EvollaConfig 設定クラス: EvollaForProteinText2Text (EvollaConfig モデル)
- Exaone4Config 設定クラス: Exaone4ForCausalLM (Exaone4Config モデル)
- ExaoneMoeConfig 設定クラス: ExaoneMoeForCausalLM (ExaoneMoeConfig モデル)
- FNetConfig 設定クラス: FNetForPreTraining (FNetConfig モデル)
- FSMTConfig 設定クラス: FSMTForConditionalGeneration (FSMTConfig モデル)
- FalconMambaConfig 設定クラス: FalconMambaForCausalLM (FalconMambaConfig モデル)
- FlaubertConfig 設定クラス: FlaubertWithLMHeadModel (FlaubertConfig モデル)
- FlavaConfig 設定クラス: FlavaForPreTraining (FlavaConfig モデル)
- Florence2Config 設定クラス: Florence2ForConditionalGeneration (Florence2Config モデル)
- FunAsrNanoConfig 設定クラス: FunAsrNanoForConditionalGeneration (FunAsrNanoConfig モデル)
- FunnelConfig 設定クラス: FunnelForPreTraining (FunnelConfig モデル)
- GPT2Config 設定クラス: GPT2LMHeadModel (GPT2Config モデル)
- GPTBigCodeConfig 設定クラス: GPTBigCodeForCausalLM (GPTBigCodeConfig モデル)
- Gemma3Config 設定クラス: Gemma3ForConditionalGeneration (Gemma3Config モデル)
- Gemma4Config 設定クラス: Gemma4ForConditionalGeneration (Gemma4Config モデル)
- Gemma4UnifiedConfig 設定クラス: Gemma4UnifiedForConditionalGeneration (Gemma4UnifiedConfig モデル)
- GlmAsrConfig 設定クラス: GlmAsrForConditionalGeneration (GlmAsrConfig モデル)
- HieraConfig 設定クラス: HieraForPreTraining (HieraConfig モデル)
- IBertConfig 設定クラス: IBertForMaskedLM (IBertConfig モデル)
- Idefics2Config 設定クラス: Idefics2ForConditionalGeneration (Idefics2Config モデル)
- Idefics3Config 設定クラス: Idefics3ForConditionalGeneration (Idefics3Config モデル)
- IdeficsConfig 設定クラス: IdeficsForVisionText2Text (IdeficsConfig モデル)
- Inkling2Config 設定クラス: Inkling2ForConditionalGeneration (Inkling2Config モデル)
- InklingConfig 設定クラス: InklingForConditionalGeneration (InklingConfig モデル)
- JanusConfig 設定クラス: JanusForConditionalGeneration (JanusConfig モデル)
- LayoutLMConfig 設定クラス: LayoutLMForMaskedLM (LayoutLMConfig モデル)
- LlavaConfig 設定クラス: LlavaForConditionalGeneration (LlavaConfig モデル)
- LlavaNextConfig 設定クラス: LlavaNextForConditionalGeneration (LlavaNextConfig モデル)
- LlavaNextVideoConfig 設定クラス: LlavaNextVideoForConditionalGeneration (LlavaNextVideoConfig モデル)
- LlavaOnevisionConfig 設定クラス: LlavaOnevisionForConditionalGeneration (LlavaOnevisionConfig モデル)
- LongformerConfig 設定クラス: LongformerForMaskedLM (LongformerConfig モデル)
- LukeConfig 設定クラス: LukeForMaskedLM (LukeConfig モデル)
- LxmertConfig 設定クラス: LxmertForPreTraining (LxmertConfig モデル)
- MPNetConfig 設定クラス: MPNetForMaskedLM (MPNetConfig モデル)
- Mamba2Config 設定クラス: Mamba2ForCausalLM (Mamba2Config モデル)
- MambaConfig 設定クラス: MambaForCausalLM (MambaConfig モデル)
- MegatronBertConfig 設定クラス: MegatronBertForPreTraining (MegatronBertConfig モデル)
- Mistral3Config 設定クラス: Mistral3ForConditionalGeneration (Mistral3Config モデル)
- Mistral4Config 設定クラス: Mistral4ForCausalLM (Mistral4Config モデル)
- MistralLarge4Config 設定クラス: MistralLarge4ForConditionalGeneration (MistralLarge4Config モデル)
- MllamaConfig 設定クラス: MllamaForConditionalGeneration (MllamaConfig モデル)
- MobileBertConfig 設定クラス: MobileBertForPreTraining (MobileBertConfig モデル)
- MossTranscribeDiarizeConfig 設定クラス: MossTranscribeDiarizeForConditionalGeneration (MossTranscribeDiarizeConfig モデル)
- MptConfig 設定クラス: MptForCausalLM (MptConfig モデル)
- MraConfig 設定クラス: MraForMaskedLM (MraConfig モデル)
- MusicFlamingoConfig 設定クラス: MusicFlamingoForConditionalGeneration (MusicFlamingoConfig モデル)
- MvpConfig 設定クラス: MvpForConditionalGeneration (MvpConfig モデル)
- NanoChatConfig 設定クラス: NanoChatForCausalLM (NanoChatConfig モデル)
- Nemotron3_5TranscribeConfig 設定クラス: Nemotron3_5TranscribeForConditionalGeneration (Nemotron3_5TranscribeConfig モデル)
- NllbMoeConfig 設定クラス: NllbMoeForConditionalGeneration (NllbMoeConfig モデル)
- OpenAIGPTConfig 設定クラス: OpenAIGPTLMHeadModel (OpenAIGPTConfig モデル)
- PaliGemmaConfig 設定クラス: PaliGemmaForConditionalGeneration (PaliGemmaConfig モデル)
- Qwen2AudioConfig 設定クラス: Qwen2AudioForConditionalGeneration (Qwen2AudioConfig モデル)
- Qwen3ASRConfig 設定クラス: Qwen3ASRForConditionalGeneration (Qwen3ASRConfig モデル)
- RoCBertConfig 設定クラス: RoCBertForPreTraining (RoCBertConfig モデル)
- RobertaConfig 設定クラス: RobertaForMaskedLM (RobertaConfig モデル)
- RobertaPreLayerNormConfig 設定クラス: RobertaPreLayerNormForMaskedLM (RobertaPreLayerNormConfig モデル)
- RwkvConfig 設定クラス: RwkvForCausalLM (RwkvConfig モデル)
- SplinterConfig 設定クラス: SplinterForPreTraining (SplinterConfig モデル)
- SqueezeBertConfig 設定クラス: SqueezeBertForMaskedLM (SqueezeBertConfig モデル)
- SwitchTransformersConfig 設定クラス: SwitchTransformersForConditionalGeneration (SwitchTransformersConfig モデル)
- T5Config 設定クラス: T5ForConditionalGeneration (T5Config モデル)
- T5Gemma2Config 設定クラス: T5Gemma2ForConditionalGeneration (T5Gemma2Config モデル)
- T5GemmaConfig 設定クラス: T5GemmaForConditionalGeneration (T5GemmaConfig モデル)
- TapasConfig 設定クラス: TapasForMaskedLM (TapasConfig モデル)
- UniSpeechConfig 設定クラス: UniSpeechForPreTraining (UniSpeechConfig モデル)
- UniSpeechSatConfig 設定クラス: UniSpeechSatForPreTraining (UniSpeechSatConfig モデル)
- ViTMAEConfig 設定クラス: ViTMAEForPreTraining (ViTMAEConfig モデル)
- VibeVoiceAsrConfig 設定クラス: VibeVoiceAsrForConditionalGeneration (VibeVoiceAsrConfig モデル)
- VideoLlavaConfig 設定クラス: VideoLlavaForConditionalGeneration (VideoLlavaConfig モデル)
- VideoMAEConfig 設定クラス: VideoMAEForPreTraining (VideoMAEConfig モデル)
- VipLlavaConfig 設定クラス: VipLlavaForConditionalGeneration (VipLlavaConfig モデル)
- VisualBertConfig 設定クラス: VisualBertForPreTraining (VisualBertConfig モデル)
- VoxtralConfig 設定クラス: VoxtralForConditionalGeneration (VoxtralConfig モデル)
- VoxtralRealtimeConfig 設定クラス: VoxtralRealtimeForConditionalGeneration (VoxtralRealtimeConfig モデル)
- Wav2Vec2Config 設定クラス: Wav2Vec2ForPreTraining (Wav2Vec2Config モデル)
- Wav2Vec2ConformerConfig 設定クラス: Wav2Vec2ConformerForPreTraining (Wav2Vec2ConformerConfig モデル)
- XLMConfig 設定クラス: XLMWithLMHeadModel (XLMConfig モデル)
- XLMRobertaConfig 設定クラス: XLMRobertaForMaskedLM (XLMRobertaConfig モデル)
- XLMRobertaXLConfig 設定クラス: XLMRobertaXLForMaskedLM (XLMRobertaXLConfig モデル)
- XLNetConfig 設定クラス: XLNetLMHeadModel (XLNetConfig モデル)
- XmodConfig 設定クラス: XmodForMaskedLM (XmodConfig モデル)
- xLSTMConfig 設定クラス: xLSTMForCausalLM (xLSTMConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(事前学習ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- albert — AlbertForPreTraining (AlbertConfig モデル)
- audioflamingo3 — AudioFlamingo3ForConditionalGeneration(AudioFlamingo3Config モデル)
- bart — BartForConditionalGeneration(BartConfigモデル)
- bert — BertForPreTraining(BertConfig モデル)
- big_bird — BigBirdForPreTraining(BigBirdConfigモデル)
- bloom — BloomForCausalLM(BloomConfig モデル)
- camembert — CamembertForMaskedLM(CamembertConfig モデル)
- colmodernvbert — ColModernVBertForRetrieval(ColModernVBertConfig モデル)
- colpali — ColPaliForRetrieval(ColPaliConfig モデル)
- colqwen2 — ColQwen2ForRetrieval(ColQwen2Config モデル)
- ctrl — CTRLLMHeadModel(CTRLConfig モデル)
- data2vec-text — Data2VecTextForMaskedLM(Data2VecTextConfig モデル)
- deberta — DebertaForMaskedLM(DebertaConfig モデル)
- deberta-v2 — DebertaV2ForMaskedLM(DebertaV2Config モデル)
- distilbert — DistilBertForMaskedLM (DistilBertConfig モデル)
- electra — ElectraForPreTraining(ElectraConfigモデル)
- ernie — ErnieForPreTraining(ErnieConfig モデル)
- evolla — EvollaForProteinText2Text(EvollaConfigモデル)
- exaone4 — Exaone4ForCausalLM(Exaone4Configモデル)
- exaone_moe — ExaoneMoeForCausalLM(ExaoneMoeConfig モデル)
- falcon_mamba — FalconMambaForCausalLM (FalconMambaConfig モデル)
- flaubert — FlaubertWithLMHeadModel(FlaubertConfig モデル)
- flava — FlavaForPreTraining(FlavaConfig モデル)
- florence2 — Florence2ForConditionalGeneration(Florence2Config モデル)
- fnet — FNetForPreTraining(FNetConfig モデル)
- fsmt — FSMTForConditionalGeneration(FSMTConfig モデル)
- fun_asr_nano — FunAsrNanoForConditionalGeneration(FunAsrNanoConfig モデル)
- funnel — FunnelForPreTraining(FunnelConfigモデル)
- gemma3 — Gemma3ForConditionalGeneration(Gemma3Config モデル)
- gemma4 — Gemma4ForConditionalGeneration(Gemma4Config モデル)
- gemma4_unified — Gemma4UnifiedForConditionalGeneration(Gemma4UnifiedConfig モデル)
- glmasr — GlmAsrForConditionalGeneration(GlmAsrConfig モデル)
- gpt-sw3 — GPT2LMHeadModel(GPT2Configモデル)
- gpt2 — GPT2LMHeadModel (GPT2Config モデル)
- gpt_bigcode — GPTBigCodeForCausalLM(GPTBigCodeConfigモデル)
- hiera — HieraForPreTraining(HieraConfigモデル)
- ibert — IBertForMaskedLM(IBertConfigモデル)
- idefics — IdeficsForVisionText2Text(IdeficsConfigモデル)
- idefics2 — Idefics2ForConditionalGeneration(Idefics2Config モデル)
- idefics3 — Idefics3ForConditionalGeneration(Idefics3Config モデル)
- inkling2 — Inkling2ForConditionalGeneration(Inkling2Configモデル)
- inkling_mm_model — InklingForConditionalGeneration(InklingConfig モデル)
- janus — JanusForConditionalGeneration(JanusConfig モデル)
- layoutlm — LayoutLMForMaskedLM(LayoutLMConfigモデル)
- llava — LlavaForConditionalGeneration(LlavaConfigモデル)
- llava_next — LlavaNextForConditionalGeneration(LlavaNextConfig モデル)
- llava_next_video — LlavaNextVideoForConditionalGeneration(LlavaNextVideoConfig モデル)
- llava_onevision — LlavaOnevisionForConditionalGeneration(LlavaOnevisionConfig モデル)
- longformer — LongformerForMaskedLM(LongformerConfig モデル)
- luke — LukeForMaskedLM (LukeConfig モデル)
- lxmert — LxmertForPreTraining(LxmertConfigモデル)
- mamba — MambaForCausalLM(MambaConfigモデル)
- mamba2 — Mamba2ForCausalLM(Mamba2Config モデル)
- megatron-bert — MegatronBertForPreTraining(MegatronBertConfig モデル)
- mistral3 — Mistral3ForConditionalGeneration(Mistral3Config モデル)
- mistral4 — Mistral4ForCausalLM(Mistral4Config モデル)
- mistral_large4 — MistralLarge4ForConditionalGeneration(MistralLarge4Config モデル)
- mllama — MllamaForConditionalGeneration(MllamaConfig モデル)
- mobilebert — MobileBertForPreTraining(MobileBertConfig モデル)
- moss_transcribe_diarize — MossTranscribeDiarizeForConditionalGeneration(MossTranscribeDiarizeConfig モデル)
- mpnet — MPNetForMaskedLM(MPNetConfig モデル)
- mpt — MptForCausalLM(MptConfig モデル)
- mra — MraForMaskedLM(MraConfigモデル)
- musicflamingo — MusicFlamingoForConditionalGeneration(MusicFlamingoConfig モデル)
- mvp — MvpForConditionalGeneration(MvpConfig モデル)
- nanochat — NanoChatForCausalLM(NanoChatConfig モデル)
- nemo_speechlm — Nemotron3_5TranscribeForConditionalGeneration(Nemotron3_5TranscribeConfig モデル)
- nllb-moe — NllbMoeForConditionalGeneration(NllbMoeConfig モデル)
- openai-gpt — OpenAIGPTLMHeadModel(OpenAIGPTConfig モデル)
- paligemma — PaliGemmaForConditionalGeneration(PaliGemmaConfig モデル)
- qwen2_audio — Qwen2AudioForConditionalGeneration(Qwen2AudioConfig モデル)
- qwen3_asr — Qwen3ASRForConditionalGeneration(Qwen3ASRConfig モデル)
- roberta — RobertaForMaskedLM(RobertaConfig モデル)
- roberta-prelayernorm — RobertaPreLayerNormForMaskedLM(RobertaPreLayerNormConfig モデル)
- roc_bert — RoCBertForPreTraining(RoCBertConfig モデル)
- rwkv — RwkvForCausalLM(RwkvConfig モデル)
- splinter — SplinterForPreTraining(SplinterConfig モデル)
- squeezebert — SqueezeBertForMaskedLM(SqueezeBertConfig モデル)
- switch_transformers — SwitchTransformersForConditionalGeneration(SwitchTransformersConfig モデル)
- t5 — T5ForConditionalGeneration(T5Config モデル)
- t5gemma — T5GemmaForConditionalGeneration(T5GemmaConfig モデル)
- t5gemma2 — T5Gemma2ForConditionalGeneration(T5Gemma2Config モデル)
- tapas — TapasForMaskedLM(TapasConfig モデル)
- unispeech — UniSpeechForPreTraining (UniSpeechConfig モデル)
- unispeech-sat — UniSpeechSatForPreTraining(UniSpeechSatConfig モデル)
- vibevoice_asr — VibeVoiceAsrForConditionalGeneration(VibeVoiceAsrConfig モデル)
- video_llava — VideoLlavaForConditionalGeneration(VideoLlavaConfigモデル)
- videomae — VideoMAEForPreTraining(VideoMAEConfig モデル)
- vipllava — VipLlavaForConditionalGeneration(VipLlavaConfig モデル)
- visual_bert — VisualBertForPreTraining(VisualBertConfig モデル)
- vit_mae — ViTMAEForPreTraining (ViTMAEConfig モデル)
- voxtral — VoxtralForConditionalGeneration(VoxtralConfig モデル)
- voxtral_realtime — VoxtralRealtimeForConditionalGeneration(VoxtralRealtimeConfig モデル)
- wav2vec2 — Wav2Vec2ForPreTraining(Wav2Vec2Configモデル)
- wav2vec2-conformer — Wav2Vec2ConformerForPreTraining(Wav2Vec2ConformerConfig モデル)
- xlm — XLMWithLMHeadModel (XLMConfig モデル)
- xlm-roberta — XLMRobertaForMaskedLM(XLMRobertaConfig モデル)
- xlm-roberta-xl — XLMRobertaXLForMaskedLM(XLMRobertaXLConfig モデル)
- xlnet — XLNetLMHeadModel(XLNetConfig モデル)
- xlstm — xLSTMForCausalLM(xLSTMConfigモデル)
- xmod — XmodForMaskedLM(XmodConfigモデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForPreTraining
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForPreTraining.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForPreTraining.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True自然言語処理
以下の自然言語処理タスクには、以下のオートクラスが利用可能です。
AutoModelForCausalLM
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスの一つ(因果言語モデリングヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- AXK1Config 設定クラス: AXK1ForCausalLM (AXK1Config モデル)
- AXK2Config 設定クラス: AXK2ForCausalLM (AXK2Config モデル)
- AfmoeConfig 設定クラス: AfmoeForCausalLM (AfmoeConfig モデル)
- ApertusConfig 設定クラス: ApertusForCausalLM(ApertusConfig モデル)
- ArceeConfig 設定クラス: ArceeForCausalLM (ArceeConfig モデル)
- AriaTextConfig 設定クラス: AriaTextForCausalLM (AriaTextConfig モデル)
- BailingHybridConfig 設定クラス: BailingHybridForCausalLM (BailingHybridConfig モデル)
- BambaConfig 設定クラス: BambaForCausalLM (BambaConfig モデル)
- BartConfig 設定クラス: BartForCausalLM (BartConfig モデル)
- BeamConfig 設定クラス: BeamForCausalLM (BeamConfig モデル)
- BertConfig 設定クラス: BertLMHeadModel (BertConfig モデル)
- BertGenerationConfig 設定クラス: BertGenerationDecoder (BertGenerationConfig モデル)
- BigBirdConfig 設定クラス: BigBirdForCausalLM (BigBirdConfig モデル)
- BigBirdPegasusConfig 設定クラス: BigBirdPegasusForCausalLM (BigBirdPegasusConfig モデル)
- BioGptConfig 設定クラス: BioGptForCausalLM (BioGptConfig モデル)
- BitNetConfig 設定クラス: BitNetForCausalLM (BitNetConfig モデル)
- BlenderbotConfig 設定クラス: BlenderbotForCausalLM (BlenderbotConfig モデル)
- BlenderbotSmallConfig 設定クラス: BlenderbotSmallForCausalLM (BlenderbotSmallConfig モデル)
- BloomConfig 設定クラス: BloomForCausalLM (BloomConfig モデル)
- BltConfig 設定クラス: BltForCausalLM (BltConfig モデル)
- CTRLConfig 設定クラス: CTRLLMHeadModel (CTRLConfig モデル)
- CamembertConfig 設定クラス: CamembertForCausalLM (CamembertConfig モデル)
- CodeGenConfig 設定クラス: CodeGenForCausalLM (CodeGenConfig モデル)
- Cohere2Config 設定クラス: Cohere2ForCausalLM (Cohere2Config モデル)
- Cohere2MoeConfig 設定クラス: Cohere2MoeForCausalLM (Cohere2MoeConfig モデル)
- CohereCompassTextConfig 設定クラス: CohereCompassForCausalLM (CohereCompassTextConfig モデル)
- CohereConfig 設定クラス: CohereForCausalLM (CohereConfig モデル)
- CpmAntConfig 設定クラス: CpmAntForCausalLM (CpmAntConfig モデル)
- CwmConfig 設定クラス: CwmForCausalLM (CwmConfig モデル)
- Data2VecTextConfig 設定クラス: Data2VecTextForCausalLM (Data2VecTextConfig モデル)
- DbrxConfig 設定クラス: DbrxForCausalLM (DbrxConfig モデル)
- DeepseekV2Config 設定クラス: DeepseekV2ForCausalLM (DeepseekV2Config モデル)
- DeepseekV32Config 設定クラス: DeepseekV32ForCausalLM (DeepseekV32Config モデル)
- DeepseekV3Config 設定クラス: DeepseekV3ForCausalLM (DeepseekV3Config モデル)
- DeepseekV4Config 設定クラス: DeepseekV4ForCausalLM (DeepseekV4Config モデル)
- DiffLlamaConfig 設定クラス: DiffLlamaForCausalLM (DiffLlamaConfig モデル)
- DogeConfig 設定クラス: DogeForCausalLM (DogeConfig モデル)
- Dots1Config 設定クラス: Dots1ForCausalLM (Dots1Config モデル)
- Dots3NoteConfig 設定クラス: Dots3NoteForCausalLM (Dots3NoteConfig モデル)
- Dots3NoteTextConfig 設定クラス: Dots3NoteForCausalLM (Dots3NoteTextConfig モデル)
- ElectraConfig 設定クラス: ElectraForCausalLM (ElectraConfig モデル)
- Emu3Config 設定クラス: Emu3ForCausalLM (Emu3Config モデル)
- Ernie4_5Config 設定クラス: Ernie4_5ForCausalLM (Ernie4_5Config モデル)
- Ernie4_5_MoeConfig 設定クラス: Ernie4_5_MoeForCausalLM (Ernie4_5_MoeConfig モデル)
- ErnieConfig 設定クラス: ErnieForCausalLM (ErnieConfig モデル)
- Exaone4Config 設定クラス: Exaone4ForCausalLM (Exaone4Config モデル)
- ExaoneMoeConfig 設定クラス: ExaoneMoeForCausalLM (ExaoneMoeConfig モデル)
- FalconConfig 設定クラス: FalconForCausalLM (FalconConfig モデル)
- FalconH1Config 設定クラス: FalconH1ForCausalLM (FalconH1Config モデル)
- FalconMambaConfig 設定クラス: FalconMambaForCausalLM (FalconMambaConfig モデル)
- FlexOlmoConfig 設定クラス: FlexOlmoForCausalLM (FlexOlmoConfig モデル)
- FlintConfig 設定クラス: FlintForCausalLM (FlintConfig モデル)
- FuyuConfig 設定クラス: FuyuForCausalLM (FuyuConfig モデル)
- GPT2Config 設定クラス: GPT2LMHeadModel (GPT2Config モデル)
- GPTBigCodeConfig 設定クラス: GPTBigCodeForCausalLM (GPTBigCodeConfig モデル)
- GPTJConfig 設定クラス: GPTJForCausalLM (GPTJConfig モデル)
- GPTNeoConfig 設定クラス: GPTNeoForCausalLM (GPTNeoConfig モデル)
- GPTNeoXConfig 設定クラス: GPTNeoXForCausalLM (GPTNeoXConfig モデル)
- GPTNeoXJapaneseConfig 設定クラス: GPTNeoXJapaneseForCausalLM (GPTNeoXJapaneseConfig モデル)
- Gemma2Config 設定クラス: Gemma2ForCausalLM (Gemma2Config モデル)
- Gemma3Config 設定クラス: Gemma3ForConditionalGeneration (Gemma3Config モデル)
- Gemma3TextConfig 設定クラス: Gemma3ForCausalLM (Gemma3TextConfig モデル)
- Gemma3nConfig 設定クラス: Gemma3nForConditionalGeneration (Gemma3nConfig モデル)
- Gemma3nTextConfig 設定クラス: Gemma3nForCausalLM (Gemma3nTextConfig モデル)
- Gemma4AssistantConfig 設定クラス: Gemma4AssistantForCausalLM (Gemma4AssistantConfig モデル)
- Gemma4Config 設定クラス: Gemma4ForConditionalGeneration (Gemma4Config モデル)
- Gemma4TextConfig 設定クラス: Gemma4ForCausalLM (Gemma4TextConfig モデル)
- Gemma4UnifiedAssistantConfig 設定クラス: Gemma4UnifiedAssistantForCausalLM (Gemma4UnifiedAssistantConfig モデル)
- Gemma4UnifiedConfig 設定クラス: Gemma4UnifiedForConditionalGeneration (Gemma4UnifiedConfig モデル)
- Gemma4UnifiedTextConfig 設定クラス: Gemma4UnifiedForCausalLM (Gemma4UnifiedTextConfig モデル)
- GemmaConfig 設定クラス: GemmaForCausalLM (GemmaConfig モデル)
- GitConfig 設定クラス: GitForCausalLM (GitConfig モデル)
- Glm4Config 設定クラス: Glm4ForCausalLM (Glm4Config モデル)
- Glm4MoeConfig 設定クラス: Glm4MoeForCausalLM (Glm4MoeConfig モデル)
- Glm4MoeLiteConfig 設定クラス: Glm4MoeLiteForCausalLM (Glm4MoeLiteConfig モデル)
- GlmConfig 設定クラス: GlmForCausalLM (GlmConfig モデル)
- GlmMoeDsaConfig 設定クラス: GlmMoeDsaForCausalLM (GlmMoeDsaConfig モデル)
- GotOcr2Config 設定クラス: GotOcr2ForConditionalGeneration (GotOcr2Config モデル)
- GptOssConfig 設定クラス: GptOssForCausalLM (GptOssConfig モデル)
- GraniteConfig 設定クラス: GraniteForCausalLM (GraniteConfig モデル)
- GraniteMoeConfig 設定クラス: GraniteMoeForCausalLM (GraniteMoeConfig モデル)
- GraniteMoeHybridConfig 設定クラス: GraniteMoeHybridForCausalLM (GraniteMoeHybridConfig モデル)
- GraniteMoeSWAConfig 設定クラス: GraniteMoeSWAForCausalLM (GraniteMoeSWAConfig モデル)
- GraniteMoeSharedConfig 設定クラス: GraniteMoeSharedForCausalLM (GraniteMoeSharedConfig モデル)
- GraniteSWAConfig 設定クラス: GraniteSWAForCausalLM (GraniteSWAConfig モデル)
- HYV3Config 設定クラス: HYV3ForCausalLM (HYV3Config モデル)
- HYV4Config 設定クラス: HYV4ForCausalLM (HYV4Config モデル)
- HeliumConfig 設定クラス: HeliumForCausalLM (HeliumConfig モデル)
- HrmTextConfig 設定クラス: HrmTextForCausalLM (HrmTextConfig モデル)
- HunYuanDenseV1Config 設定クラス: HunYuanDenseV1ForCausalLM (HunYuanDenseV1Config モデル)
- HunYuanMoEV1Config 設定クラス: HunYuanMoEV1ForCausalLM (HunYuanMoEV1Config モデル)
- HyperCLOVAXConfig 設定クラス: HyperCLOVAXForCausalLM (HyperCLOVAXConfig モデル)
- IQuestCoderConfig 設定クラス: LlamaForCausalLM (IQuestCoderConfig モデル)
- Inkling2TextConfig 設定クラス: Inkling2ForCausalLM (Inkling2TextConfig モデル)
- InklingTextConfig 設定クラス: InklingForCausalLM (InklingTextConfig モデル)
- InternLM3Config 設定クラス: LlamaForCausalLM (InternLM3Config モデル)
- Jais2Config 設定クラス: Jais2ForCausalLM (Jais2Config モデル)
- JambaConfig 設定クラス: JambaForCausalLM (JambaConfig モデル)
- JetMoeConfig 設定クラス: JetMoeForCausalLM (JetMoeConfig モデル)
- KimiLinearConfig 設定クラス: KimiLinearForCausalLM (KimiLinearConfig モデル)
- Kolibri1Config 設定クラス: Kolibri1ForCausalLM (Kolibri1Config モデル)
- LagunaConfig 設定クラス: LagunaForCausalLM (LagunaConfig モデル)
- Lfm2Config 設定クラス: Lfm2ForCausalLM (Lfm2Config モデル)
- Lfm2MoeConfig 設定クラス: Lfm2MoeForCausalLM (Lfm2MoeConfig モデル)
- Lfm3Config 設定クラス: Lfm3ForCausalLM (Lfm3Config モデル)
- Llama4Config 設定クラス: Llama4ForCausalLM (Llama4Config モデル)
- Llama4TextConfig 設定クラス: Llama4ForCausalLM (Llama4TextConfig モデル)
- LlamaConfig 設定クラス: LlamaForCausalLM (LlamaConfig モデル)
- LongcatFlashConfig 設定クラス: LongcatFlashForCausalLM(LongcatFlashConfig モデル)
- MBartConfig 設定クラス: MBartForCausalLM (MBartConfig モデル)
- Mamba2Config 設定クラス: Mamba2ForCausalLM (Mamba2Config モデル)
- MambaConfig 設定クラス: MambaForCausalLM (MambaConfig モデル)
- MarianConfig 設定クラス: MarianForCausalLM (MarianConfig モデル)
- MegatronBertConfig 設定クラス: MegatronBertForCausalLM (MegatronBertConfig モデル)
- MellumConfig 設定クラス: MellumForCausalLM (MellumConfig モデル)
- MiMoV2FlashConfig 設定クラス: MiMoV2FlashForCausalLM (MiMoV2FlashConfig モデル)
- MiniCPM3Config 設定クラス: MiniCPM3ForCausalLM (MiniCPM3Config モデル)
- MiniMaxConfig 設定クラス: MiniMaxForCausalLM (MiniMaxConfig モデル)
- MiniMaxM2Config 設定クラス: MiniMaxM2ForCausalLM (MiniMaxM2Config モデル)
- MiniMaxM3VLTextConfig 設定クラス: MiniMaxM3VLForCausalLM (MiniMaxM3VLTextConfig モデル)
- Ministral3Config 設定クラス: Ministral3ForCausalLM (Ministral3Config モデル)
- MinistralConfig 設定クラス: MinistralForCausalLM (MinistralConfig モデル)
- MistralConfig 設定クラス: MistralForCausalLM (MistralConfig モデル)
- MixtralConfig 設定クラス: MixtralForCausalLM (MixtralConfig モデル)
- MllamaConfig 設定クラス: MllamaForCausalLM (MllamaConfig モデル)
- ModernBertDecoderConfig 設定クラス: ModernBertDecoderForCausalLM (ModernBertDecoderConfig モデル)
- MoshiConfig 設定クラス: MoshiForCausalLM (MoshiConfig モデル)
- MossTranscribeDiarizeConfig 設定クラス: MossTranscribeDiarizeForConditionalGeneration (MossTranscribeDiarizeConfig モデル)
- MptConfig 設定クラス: MptForCausalLM (MptConfig モデル)
- MusicgenConfig 設定クラス: MusicgenForCausalLM (MusicgenConfig モデル)
- MusicgenMelodyConfig 設定クラス: MusicgenMelodyForCausalLM (MusicgenMelodyConfig モデル)
- MvpConfig 設定クラス: MvpForCausalLM (MvpConfig モデル)
- NanoChatConfig 設定クラス: NanoChatForCausalLM (NanoChatConfig モデル)
- NemotronConfig 設定クラス: NemotronForCausalLM (NemotronConfig モデル)
- NemotronHConfig 設定クラス: NemotronHForCausalLM (NemotronHConfig モデル)
- OPTConfig 設定クラス: OPTForCausalLM (OPTConfig モデル)
- Olmo2Config 設定クラス: Olmo2ForCausalLM (Olmo2Config モデル)
- Olmo3Config 設定クラス: Olmo3ForCausalLM (Olmo3Config モデル)
- OlmoConfig 設定クラス: OlmoForCausalLM (OlmoConfig モデル)
- OlmoHybridConfig 設定クラス: OlmoHybridForCausalLM (OlmoHybridConfig モデル)
- OlmoeConfig 設定クラス: OlmoeForCausalLM (OlmoeConfig モデル)
- OpenAIGPTConfig 設定クラス: OpenAIGPTLMHeadModel (OpenAIGPTConfig モデル)
- PLBartConfig 設定クラス: PLBartForCausalLM (PLBartConfig モデル)
- PegasusConfig 設定クラス: PegasusForCausalLM (PegasusConfig モデル)
- PersimmonConfig 設定クラス: PersimmonForCausalLM (PersimmonConfig モデル)
- Phi3Config 設定クラス: Phi3ForCausalLM (Phi3Config モデル)
- Phi4MultimodalConfig 設定クラス: Phi4MultimodalForCausalLM (Phi4MultimodalConfig モデル)
- PhiConfig 設定クラス: PhiForCausalLM (PhiConfig モデル)
- PhimoeConfig 設定クラス: PhimoeForCausalLM (PhimoeConfig モデル)
- ProphetNetConfig 設定クラス: ProphetNetForCausalLM (ProphetNetConfig モデル)
- Qwen2Config 設定クラス: Qwen2ForCausalLM (Qwen2Config モデル)
- Qwen2MoeConfig 設定クラス: Qwen2MoeForCausalLM (Qwen2MoeConfig モデル)
- Qwen3Config 設定クラス: Qwen3ForCausalLM (Qwen3Config モデル)
- Qwen3MoeConfig 設定クラス: Qwen3MoeForCausalLM (Qwen3MoeConfig モデル)
- Qwen3NextConfig 設定クラス: Qwen3NextForCausalLM (Qwen3NextConfig モデル)
- Qwen3_5Config 設定クラス: Qwen3_5ForCausalLM (Qwen3_5Config モデル)
- Qwen3_5MoeConfig 設定クラス: Qwen3_5MoeForCausalLM (Qwen3_5MoeConfig モデル)
- Qwen3_5MoeTextConfig 設定クラス: Qwen3_5MoeForCausalLM (Qwen3_5MoeTextConfig モデル)
- Qwen3_5TextConfig 設定クラス: Qwen3_5ForCausalLM (Qwen3_5TextConfig モデル)
- Qwen4ExpConfig 設定クラス: Qwen4ExpForCausalLM (Qwen4ExpConfig モデル)
- Qwen4ExpTextConfig 設定クラス: Qwen4ExpForCausalLM (Qwen4ExpTextConfig モデル)
- RecurrentGemmaConfig 設定クラス: RecurrentGemmaForCausalLM (RecurrentGemmaConfig モデル)
- ReformerConfig 設定クラス: ReformerModelWithLMHead (ReformerConfig モデル)
- RemBertConfig 設定クラス: RemBertForCausalLM (RemBertConfig モデル)
- RoCBertConfig 設定クラス: RoCBertForCausalLM (RoCBertConfig モデル)
- RoFormerConfig 設定クラス: RoFormerForCausalLM (RoFormerConfig モデル)
- RobertaConfig 設定クラス: RobertaForCausalLM (RobertaConfig モデル)
- RobertaPreLayerNormConfig 設定クラス: RobertaPreLayerNormForCausalLM (RobertaPreLayerNormConfig モデル)
- RwkvConfig 設定クラス: RwkvForCausalLM (RwkvConfig モデル)
- SeedOssConfig 設定クラス: SeedOssForCausalLM (SeedOssConfig モデル)
- SmolLM3Config 設定クラス: SmolLM3ForCausalLM (SmolLM3Config モデル)
- SolarOpenConfig 設定クラス: SolarOpenForCausalLM (SolarOpenConfig モデル)
- StableLmConfig 設定クラス: StableLmForCausalLM (StableLmConfig モデル)
- Starcoder2Config 設定クラス: Starcoder2ForCausalLM (Starcoder2Config モデル)
- TrOCRConfig 設定クラス: TrOCRForCausalLM (TrOCRConfig モデル)
- VaultGemmaConfig 設定クラス: VaultGemmaForCausalLM (VaultGemmaConfig モデル)
- WhisperConfig 設定クラス: WhisperForCausalLM (WhisperConfig モデル)
- XGLMConfig 設定クラス: XGLMForCausalLM (XGLMConfig モデル)
- XLMConfig 設定クラス: XLMWithLMHeadModel (XLMConfig モデル)
- XLMRobertaConfig 設定クラス: XLMRobertaForCausalLM (XLMRobertaConfig モデル)
- XLMRobertaXLConfig 設定クラス: XLMRobertaXLForCausalLM (XLMRobertaXLConfig モデル)
- XLNetConfig 設定クラス: XLNetLMHeadModel (XLNetConfig モデル)
- XmodConfig 設定クラス: XmodForCausalLM (XmodConfig モデル)
- YoutuConfig 設定クラス: YoutuForCausalLM (YoutuConfig モデル)
- Zamba2Config 設定クラス: Zamba2ForCausalLM (Zamba2Config モデル)
- ZambaConfig 設定クラス: ZambaForCausalLM (ZambaConfig モデル)
- ZayaConfig 設定クラス: ZayaForCausalLM (ZayaConfig モデル)
- xLSTMConfig 設定クラス: xLSTMForCausalLM (xLSTMConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(因果言語モデリングヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- afmoe — AfmoeForCausalLM (AfmoeConfig モデル)
- apertus — ApertusForCausalLM (ApertusConfig モデル)
- arcee — ArceeForCausalLM (ArceeConfig モデル)
- aria_text — AriaTextForCausalLM(AriaTextConfig モデル)
- axk1 — AXK1ForCausalLM(AXK1Config モデル)
- axk2 — AXK2ForCausalLM(AXK2Config モデル)
- bailing_hybrid — BailingHybridForCausalLM(BailingHybridConfig モデル)
- bamba — BambaForCausalLM (BambaConfig モデル)
- bart — BartForCausalLM(BartConfigモデル)
- beam — BeamForCausalLM(BeamConfig モデル)
- bert — BertLMHeadModel (BertConfig モデル)
- bert-generation — BertGenerationDecoder(BertGenerationConfig モデル)
- big_bird — BigBirdForCausalLM(BigBirdConfigモデル)
- bigbird_pegasus — BigBirdPegasusForCausalLM(BigBirdPegasusConfig モデル)
- biogpt — BioGptForCausalLM (BioGptConfig モデル)
- bitnet — BitNetForCausalLM(BitNetConfig モデル)
- blenderbot — BlenderbotForCausalLM(BlenderbotConfig モデル)
- blenderbot-small — BlenderbotSmallForCausalLM(BlenderbotSmallConfig モデル)
- bloom — BloomForCausalLM(BloomConfig モデル)
- blt — BltForCausalLM(BltConfig モデル)
- camembert — CamembertForCausalLM(CamembertConfig モデル)
- codegen — CodeGenForCausalLM(CodeGenConfig モデル)
- cohere — CohereForCausalLM(CohereConfigモデル)
- cohere2 — Cohere2ForCausalLM (Cohere2Config モデル)
- cohere2_moe — Cohere2MoeForCausalLM(Cohere2MoeConfig モデル)
- cohere_compass_text — CohereCompassForCausalLM(CohereCompassTextConfig モデル)
- cpmant — CpmAntForCausalLM(CpmAntConfigモデル)
- ctrl — CTRLLMHeadModel(CTRLConfig モデル)
- cwm — CwmForCausalLM(CwmConfigモデル)
- data2vec-text — Data2VecTextForCausalLM(Data2VecTextConfig モデル)
- dbrx — DbrxForCausalLM(DbrxConfig モデル)
- deepseek_v2 — DeepseekV2ForCausalLM(DeepseekV2Config モデル)
- deepseek_v3 — DeepseekV3ForCausalLM (DeepseekV3Config モデル)
- deepseek_v32 — DeepseekV32ForCausalLM(DeepseekV32Config モデル)
- deepseek_v4 — DeepseekV4ForCausalLM (DeepseekV4Config モデル)
- diffllama — DiffLlamaForCausalLM(DiffLlamaConfigモデル)
- doge — DogeForCausalLM(DogeConfigモデル)
- dots1 — Dots1ForCausalLM(Dots1Config モデル)
- dots3_note — Dots3NoteForCausalLM(Dots3NoteConfigモデル)
- dots3_note_text — Dots3NoteForCausalLM (Dots3NoteTextConfig モデル)
- electra — ElectraForCausalLM (ElectraConfig モデル)
- emu3 — Emu3ForCausalLM(Emu3Config モデル)
- ernie — ErnieForCausalLM(ErnieConfig モデル)
- ernie4_5 — Ernie4_5ForCausalLM(Ernie4_5Config モデル)
- ernie4_5_moe — Ernie4_5_MoeForCausalLM(Ernie4_5_MoeConfig モデル)
- exaone4 — Exaone4ForCausalLM(Exaone4Configモデル)
- exaone_moe — ExaoneMoeForCausalLM(ExaoneMoeConfig モデル)
- falcon — FalconForCausalLM (FalconConfig モデル)
- falcon_h1 — FalconH1ForCausalLM(FalconH1Config モデル)
- falcon_mamba — FalconMambaForCausalLM (FalconMambaConfig モデル)
- flex_olmo — FlexOlmoForCausalLM(FlexOlmoConfig モデル)
- flint — FlintForCausalLM(FlintConfigモデル)
- fuyu — FuyuForCausalLM(FuyuConfigモデル)
- gemma — GemmaForCausalLM(GemmaConfig モデル)
- gemma2 — Gemma2ForCausalLM(Gemma2Config モデル)
- gemma3 — Gemma3ForConditionalGeneration(Gemma3Config モデル)
- gemma3_text — Gemma3ForCausalLM(Gemma3TextConfig モデル)
- gemma3n — Gemma3nForConditionalGeneration(Gemma3nConfig モデル)
- gemma3n_text — Gemma3nForCausalLM(Gemma3nTextConfig モデル)
- gemma4 — Gemma4ForConditionalGeneration(Gemma4Config モデル)
- gemma4_assistant — Gemma4AssistantForCausalLM(Gemma4AssistantConfig モデル)
- gemma4_text — Gemma4ForCausalLM(Gemma4TextConfig モデル)
- gemma4_unified — Gemma4UnifiedForConditionalGeneration(Gemma4UnifiedConfig モデル)
- gemma4_unified_assistant — Gemma4UnifiedAssistantForCausalLM(Gemma4UnifiedAssistantConfig モデル)
- gemma4_unified_text — Gemma4UnifiedForCausalLM(Gemma4UnifiedTextConfig モデル)
- git — GitForCausalLM(GitConfigモデル)
- glm — GlmForCausalLM(GlmConfig モデル)
- glm4 — Glm4ForCausalLM(Glm4Config モデル)
- glm4_moe — Glm4MoeForCausalLM(Glm4MoeConfig モデル)
- glm4_moe_lite — Glm4MoeLiteForCausalLM(Glm4MoeLiteConfig モデル)
- glm_moe_dsa — GlmMoeDsaForCausalLM(GlmMoeDsaConfig モデル)
- got_ocr2 — GotOcr2ForConditionalGeneration(GotOcr2Config モデル)
- gpt-sw3 — GPT2LMHeadModel(GPT2Configモデル)
- gpt2 — GPT2LMHeadModel (GPT2Config モデル)
- gpt_bigcode — GPTBigCodeForCausalLM(GPTBigCodeConfigモデル)
- gpt_neo — GPTNeoForCausalLM(GPTNeoConfigモデル)
- gpt_neox — GPTNeoXForCausalLM(GPTNeoXConfigモデル)
- gpt_neox_japanese — GPTNeoXJapaneseForCausalLM(GPTNeoXJapaneseConfig モデル)
- gpt_oss — GptOssForCausalLM(GptOssConfig モデル)
- gptj — GPTJForCausalLM(GPTJConfig モデル)
- granite — GraniteForCausalLM(GraniteConfigモデル)
- granite_swa — GraniteSWAForCausalLM(GraniteSWAConfigモデル)
- granitemoe — GraniteMoeForCausalLM(GraniteMoeConfigモデル)
- granitemoe_swa — GraniteMoeSWAForCausalLM(GraniteMoeSWAConfig モデル)
- granitemoehybrid — GraniteMoeHybridForCausalLM(GraniteMoeHybridConfig モデル)
- granitemoeshared — GraniteMoeSharedForCausalLM(GraniteMoeSharedConfig モデル)
- helium — HeliumForCausalLM(HeliumConfigモデル)
- hrm_text — HrmTextForCausalLM(HrmTextConfig モデル)
- hunyuan_v1_dense — HunYuanDenseV1ForCausalLM(HunYuanDenseV1Config モデル)
- hunyuan_v1_moe — HunYuanMoEV1ForCausalLM(HunYuanMoEV1Config モデル)
- hy_v3 — HYV3ForCausalLM(HYV3Config モデル)
- hy_v4 — HYV4ForCausalLM(HYV4Config モデル)
- hyperclovax — HyperCLOVAXForCausalLM(HyperCLOVAXConfigモデル)
- inkling2_text — Inkling2ForCausalLM(Inkling2TextConfig モデル)
- inkling_text — InklingForCausalLM(InklingTextConfig モデル)
- internlm3 — LlamaForCausalLM (InternLM3Config モデル)
- iquestcoder — LlamaForCausalLM(IQuestCoderConfigモデル)
- jais2 — Jais2ForCausalLM(Jais2Config モデル)
- jamba — JambaForCausalLM(JambaConfig モデル)
- jetmoe — JetMoeForCausalLM(JetMoeConfigモデル)
- kimi_linear — KimiLinearForCausalLM(KimiLinearConfig モデル)
- kolibri1 — Kolibri1ForCausalLM(Kolibri1Config モデル)
- laguna — LagunaForCausalLM(LagunaConfig モデル)
- lfm2 — Lfm2ForCausalLM(Lfm2Config モデル)
- lfm2_moe — Lfm2MoeForCausalLM(Lfm2MoeConfig モデル)
- lfm3 — Lfm3ForCausalLM(Lfm3Configモデル)
- llama — LlamaForCausalLM (LlamaConfig モデル)
- llama4 — Llama4ForCausalLM(Llama4Config モデル)
- llama4_text — Llama4ForCausalLM(Llama4TextConfig モデル)
- longcat_flash — LongcatFlashForCausalLM(LongcatFlashConfig モデル)
- mamba — MambaForCausalLM(MambaConfigモデル)
- mamba2 — Mamba2ForCausalLM(Mamba2Config モデル)
- marian — MarianForCausalLM(MarianConfig モデル)
- mbart — MBartForCausalLM(MBartConfig モデル)
- megatron-bert — MegatronBertForCausalLM (MegatronBertConfig モデル)
- mellum — MellumForCausalLM (MellumConfig モデル)
- mimo_v2_flash — MiMoV2FlashForCausalLM(MiMoV2FlashConfig モデル)
- minicpm3 — MiniCPM3ForCausalLM(MiniCPM3Config モデル)
- minimax — MiniMaxForCausalLM(MiniMaxConfig モデル)
- minimax_m2 — MiniMaxM2ForCausalLM(MiniMaxM2Config モデル)
- minimax_m3_vl_text — MiniMaxM3VLForCausalLM(MiniMaxM3VLTextConfig モデル)
- ministral — MinistralForCausalLM(MinistralConfig モデル)
- ministral3 — Ministral3ForCausalLM(Ministral3Config モデル)
- mistral — MistralForCausalLM(MistralConfig モデル)
- mixtral — MixtralForCausalLM(MixtralConfig モデル)
- mllama — MllamaForCausalLM(MllamaConfig モデル)
- modernbert-decoder — ModernBertDecoderForCausalLM(ModernBertDecoderConfig モデル)
- moshi — MoshiForCausalLM(MoshiConfigモデル)
- moss_transcribe_diarize — MossTranscribeDiarizeForConditionalGeneration(MossTranscribeDiarizeConfig モデル)
- mpt — MptForCausalLM(MptConfig モデル)
- musicgen — MusicgenForCausalLM(MusicgenConfig モデル)
- musicgen_melody — MusicgenMelodyForCausalLM(MusicgenMelodyConfig モデル)
- mvp — MvpForCausalLM(MvpConfig モデル)
- nanochat — NanoChatForCausalLM(NanoChatConfig モデル)
- nemotron — NemotronForCausalLM(NemotronConfigモデル)
- nemotron_h — NemotronHForCausalLM(NemotronHConfig モデル)
- olmo — OlmoForCausalLM(OlmoConfig モデル)
- olmo2 — Olmo2ForCausalLM(Olmo2Config モデル)
- olmo3 — Olmo3ForCausalLM(Olmo3Config モデル)
- olmo_hybrid — OlmoHybridForCausalLM(OlmoHybridConfig モデル)
- olmoe — OlmoeForCausalLM(OlmoeConfig モデル)
- openai-gpt — OpenAIGPTLMHeadModel(OpenAIGPTConfig モデル)
- opt — OPTForCausalLM(OPTConfig モデル)
- pegasus — PegasusForCausalLM(PegasusConfig モデル)
- persimmon — PersimmonForCausalLM(PersimmonConfig モデル)
- phi — PhiForCausalLM(PhiConfig モデル)
- phi3 — Phi3ForCausalLM(Phi3Config モデル)
- phi4_multimodal — Phi4MultimodalForCausalLM(Phi4MultimodalConfig モデル)
- phimoe — PhimoeForCausalLM (PhimoeConfig モデル)
- plbart — PLBartForCausalLM(PLBartConfig モデル)
- prophetnet — ProphetNetForCausalLM (ProphetNetConfig モデル)
- qwen2 — Qwen2ForCausalLM (Qwen2Config モデル)
- qwen2_moe — Qwen2MoeForCausalLM(Qwen2MoeConfig モデル)
- qwen3 — Qwen3ForCausalLM(Qwen3Configモデル)
- qwen3_5 — Qwen3_5ForCausalLM(Qwen3_5Config モデル)
- qwen3_5_moe — Qwen3_5MoeForCausalLM(Qwen3_5MoeConfig モデル)
- qwen3_5_moe_text — Qwen3_5MoeForCausalLM(Qwen3_5MoeTextConfig モデル)
- qwen3_5_text — Qwen3_5ForCausalLM(Qwen3_5TextConfig モデル)
- qwen3_moe — Qwen3MoeForCausalLM(Qwen3MoeConfig モデル)
- qwen3_next — Qwen3NextForCausalLM(Qwen3NextConfig モデル)
- qwen4_exp — Qwen4ExpForCausalLM(Qwen4ExpConfig モデル)
- qwen4_exp_text — Qwen4ExpForCausalLM(Qwen4ExpTextConfig モデル)
- recurrent_gemma — RecurrentGemmaForCausalLM(RecurrentGemmaConfig モデル)
- reformer — ReformerModelWithLMHead(ReformerConfig モデル)
- rembert — RemBertForCausalLM(RemBertConfig モデル)
- roberta — RobertaForCausalLM(RobertaConfig モデル)
- roberta-prelayernorm — RobertaPreLayerNormForCausalLM(RobertaPreLayerNormConfig モデル)
- roc_bert — RoCBertForCausalLM(RoCBertConfig モデル)
- roformer — RoFormerForCausalLM(RoFormerConfig モデル)
- rwkv — RwkvForCausalLM(RwkvConfig モデル)
- seed_oss — SeedOssForCausalLM(SeedOssConfig モデル)
- smollm3 — SmolLM3ForCausalLM(SmolLM3Config モデル)
- solar_open — SolarOpenForCausalLM(SolarOpenConfig モデル)
- stablelm — StableLmForCausalLM(StableLmConfig モデル)
- starcoder2 — Starcoder2ForCausalLM(Starcoder2Config モデル)
- trocr — TrOCRForCausalLM (TrOCRConfig モデル)
- vaultgemma — VaultGemmaForCausalLM(VaultGemmaConfig モデル)
- whisper — WhisperForCausalLM (WhisperConfig モデル)
- xglm — XGLMForCausalLM(XGLMConfig モデル)
- xlm — XLMWithLMHeadModel (XLMConfig モデル)
- xlm-roberta — XLMRobertaForCausalLM (XLMRobertaConfig モデル)
- xlm-roberta-xl — XLMRobertaXLForCausalLM(XLMRobertaXLConfig モデル)
- xlnet — XLNetLMHeadModel(XLNetConfig モデル)
- xlstm — xLSTMForCausalLM(xLSTMConfigモデル)
- xmod — XmodForCausalLM(XmodConfig モデル)
- youtu — YoutuForCausalLM(YoutuConfigモデル)
- zamba — ZambaForCausalLM(ZambaConfig モデル)
- zamba2 — Zamba2ForCausalLM(Zamba2Config モデル)
- zaya — ZayaForCausalLM(ZayaConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForCausalLM
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForCausalLM.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForCausalLM.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForMaskedLM
これは汎用的なモデルクラスであり、from_pretrained() クラスメソッドまたは from_config() クラスメソッドで生成された場合、マスク言語モデリングヘッドを備えたライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- AlbertConfig 設定クラス: AlbertForMaskedLM (AlbertConfig モデル)
- BartConfig 設定クラス: BartForConditionalGeneration (BartConfig モデル)
- BertConfig 設定クラス: BertForMaskedLM (BertConfig モデル)
- BigBirdConfig 設定クラス: BigBirdForMaskedLM (BigBirdConfig モデル)
- CamembertConfig 設定クラス: CamembertForMaskedLM (CamembertConfig モデル)
- ConvBertConfig 設定クラス: ConvBertForMaskedLM (ConvBertConfig モデル)
- Data2VecTextConfig 設定クラス: Data2VecTextForMaskedLM (Data2VecTextConfig モデル)
- DebertaConfig 設定クラス: DebertaForMaskedLM (DebertaConfig モデル)
- DebertaV2Config 設定クラス: DebertaV2ForMaskedLM (DebertaV2Config モデル)
- DistilBertConfig 設定クラス: DistilBertForMaskedLM (DistilBertConfig モデル)
- ElectraConfig 設定クラス: ElectraForMaskedLM (ElectraConfig モデル)
- ErnieConfig 設定クラス: ErnieForMaskedLM (ErnieConfig モデル)
- EsmConfig 設定クラス: EsmForMaskedLM (EsmConfig モデル)
- EsmcConfig 設定クラス: EsmcForMaskedLM (EsmcConfig モデル)
- EuroBertConfig 設定クラス: EuroBertForMaskedLM (EuroBertConfig モデル)
- FNetConfig 設定クラス: FNetForMaskedLM (FNetConfig モデル)
- FlaubertConfig 設定クラス: FlaubertWithLMHeadModel (FlaubertConfig モデル)
- FunnelConfig 設定クラス: FunnelForMaskedLM (FunnelConfig モデル)
- GteConfig 設定クラス: GteForMaskedLM (GteConfig モデル)
- IBertConfig 設定クラス: IBertForMaskedLM (IBertConfig モデル)
- JinaEmbeddingsV3Config 設定クラス: JinaEmbeddingsV3ForMaskedLM (JinaEmbeddingsV3Config モデル)
- LayoutLMConfig 設定クラス: LayoutLMForMaskedLM (LayoutLMConfig モデル)
- LongformerConfig 設定クラス: LongformerForMaskedLM (LongformerConfig モデル)
- LukeConfig 設定クラス: LukeForMaskedLM (LukeConfig モデル)
- MBartConfig 設定クラス: MBartForConditionalGeneration (MBartConfig モデル)
- MPNetConfig 設定クラス: MPNetForMaskedLM (MPNetConfig モデル)
- MegatronBertConfig 設定クラス: MegatronBertForMaskedLM (MegatronBertConfig モデル)
- MobileBertConfig 設定クラス: MobileBertForMaskedLM (MobileBertConfig モデル)
- ModernBertConfig 設定クラス: ModernBertForMaskedLM (ModernBertConfig モデル)
- ModernVBertConfig 設定クラス: ModernVBertForMaskedLM (ModernVBertConfig モデル)
- MraConfig 設定クラス: MraForMaskedLM (MraConfig モデル)
- MvpConfig 設定クラス: MvpForConditionalGeneration (MvpConfig モデル)
- NeoMMEConfig 設定クラス: NeoMMEForMaskedLM (NeoMMEConfig モデル)
- NomicBertConfig 設定クラス: NomicBertForMaskedLM (NomicBertConfig モデル)
- NystromformerConfig 設定クラス: NystromformerForMaskedLM (NystromformerConfig モデル)
- PerceiverConfig 設定クラス: PerceiverForMaskedLM (PerceiverConfig モデル)
- ReformerConfig 設定クラス: ReformerForMaskedLM (ReformerConfig モデル)
- RemBertConfig 設定クラス: RemBertForMaskedLM (RemBertConfig モデル)
- RoCBertConfig 設定クラス: RoCBertForMaskedLM (RoCBertConfig モデル)
- RoFormerConfig 設定クラス: RoFormerForMaskedLM (RoFormerConfig モデル)
- RobertaConfig 設定クラス: RobertaForMaskedLM (RobertaConfig モデル)
- RobertaPreLayerNormConfig 設定クラス: RobertaPreLayerNormForMaskedLM (RobertaPreLayerNormConfig モデル)
- SqueezeBertConfig 設定クラス: SqueezeBertForMaskedLM (SqueezeBertConfig モデル)
- TapasConfig 設定クラス: TapasForMaskedLM (TapasConfig モデル)
- UltraVBertConfig 設定クラス: UltraVBertForMaskedLM (UltraVBertConfig モデル)
- UltrabertConfig 設定クラス: UltrabertForMaskedLM (UltrabertConfig モデル)
- XLMConfig 設定クラス: XLMWithLMHeadModel (XLMConfig モデル)
- XLMRobertaConfig 設定クラス: XLMRobertaForMaskedLM (XLMRobertaConfig モデル)
- XLMRobertaXLConfig 設定クラス: XLMRobertaXLForMaskedLM (XLMRobertaXLConfig モデル)
- XmodConfig 設定クラス: XmodForMaskedLM (XmodConfig モデル)
- YosoConfig 設定クラス: YosoForMaskedLM (YosoConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(マスクド言語モデリングヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- albert — AlbertForMaskedLM(AlbertConfig モデル)
- bart — BartForConditionalGeneration(BartConfigモデル)
- bert — BertForMaskedLM(BertConfig モデル)
- big_bird — BigBirdForMaskedLM(BigBirdConfig モデル)
- camembert — CamembertForMaskedLM(CamembertConfig モデル)
- convbert — ConvBertForMaskedLM (ConvBertConfig モデル)
- data2vec-text — Data2VecTextForMaskedLM(Data2VecTextConfig モデル)
- deberta — DebertaForMaskedLM(DebertaConfig モデル)
- deberta-v2 — DebertaV2ForMaskedLM(DebertaV2Config モデル)
- distilbert — DistilBertForMaskedLM (DistilBertConfig モデル)
- electra — ElectraForMaskedLM(ElectraConfig モデル)
- ernie — ErnieForMaskedLM(ErnieConfig モデル)
- esm — EsmForMaskedLM(EsmConfigモデル)
- esmc — EsmcForMaskedLM(EsmcConfig モデル)
- eurobert — EuroBertForMaskedLM(EuroBertConfigモデル)
- flaubert — FlaubertWithLMHeadModel(FlaubertConfig モデル)
- fnet — FNetForMaskedLM(FNetConfig モデル)
- funnel — FunnelForMaskedLM(FunnelConfig モデル)
- gte — GteForMaskedLM(GteConfigモデル)
- ibert — IBertForMaskedLM(IBertConfigモデル)
- jina_embeddings_v3 — JinaEmbeddingsV3ForMaskedLM(JinaEmbeddingsV3Config モデル)
- layoutlm — LayoutLMForMaskedLM(LayoutLMConfigモデル)
- longformer — LongformerForMaskedLM(LongformerConfig モデル)
- luke — LukeForMaskedLM (LukeConfig モデル)
- mbart — MBartForConditionalGeneration(MBartConfig モデル)
- megatron-bert — MegatronBertForMaskedLM (MegatronBertConfig モデル)
- mobilebert — MobileBertForMaskedLM(MobileBertConfig モデル)
- modernbert — ModernBertForMaskedLM(ModernBertConfig モデル)
- modernvbert — ModernVBertForMaskedLM(ModernVBertConfig モデル)
- mpnet — MPNetForMaskedLM(MPNetConfig モデル)
- mra — MraForMaskedLM(MraConfigモデル)
- mvp — MvpForConditionalGeneration(MvpConfig モデル)
- neomme — NeoMMEForMaskedLM (NeoMMEConfig モデル)
- nomic_bert — NomicBertForMaskedLM(NomicBertConfig モデル)
- nystromformer — NystromformerForMaskedLM(NystromformerConfig モデル)
- perceiver — PerceiverForMaskedLM(PerceiverConfigモデル)
- reformer — ReformerForMaskedLM(ReformerConfigモデル)
- rembert — RemBertForMaskedLM (RemBertConfig モデル)
- roberta — RobertaForMaskedLM(RobertaConfig モデル)
- roberta-prelayernorm — RobertaPreLayerNormForMaskedLM(RobertaPreLayerNormConfig モデル)
- roc_bert — RoCBertForMaskedLM(RoCBertConfig モデル)
- roformer — RoFormerForMaskedLM(RoFormerConfig モデル)
- squeezebert — SqueezeBertForMaskedLM(SqueezeBertConfig モデル)
- tapas — TapasForMaskedLM(TapasConfig モデル)
- ultrabert — UltrabertForMaskedLM(UltrabertConfig モデル)
- ultravbert — UltraVBertForMaskedLM(UltraVBertConfig モデル)
- xlm — XLMWithLMHeadModel (XLMConfig モデル)
- xlm-roberta — XLMRobertaForMaskedLM(XLMRobertaConfig モデル)
- xlm-roberta-xl — XLMRobertaXLForMaskedLM(XLMRobertaXLConfig モデル)
- xmod — XmodForMaskedLM(XmodConfigモデル)
- yoso — YosoForMaskedLM (YosoConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForMaskedLM
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForMaskedLM.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForMaskedLM.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForMaskGeneration
AutoModelForSeq2SeqLM
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスのいずれか(シーケンスツーシーケンス言語モデリングヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- AudioFlamingo3Config 設定クラス: AudioFlamingo3ForConditionalGeneration (AudioFlamingo3Config モデル)
- BartConfig 設定クラス: BartForConditionalGeneration (BartConfig モデル)
- BigBirdPegasusConfig 設定クラス: BigBirdPegasusForConditionalGeneration (BigBirdPegasusConfig モデル)
- BlenderbotConfig 設定クラス: BlenderbotForConditionalGeneration (BlenderbotConfig モデル)
- BlenderbotSmallConfig 設定クラス: BlenderbotSmallForConditionalGeneration (BlenderbotSmallConfig モデル)
- EncoderDecoderConfig 設定クラス: EncoderDecoderModel (EncoderDecoderConfig モデル)
- FSMTConfig 設定クラス: FSMTForConditionalGeneration (FSMTConfig モデル)
- GlmAsrConfig 設定クラス: GlmAsrForConditionalGeneration (GlmAsrConfig モデル)
- GraniteSpeechConfig 設定クラス: GraniteSpeechForConditionalGeneration (GraniteSpeechConfig モデル)
- GraniteSpeechPlusConfig 設定クラス: GraniteSpeechPlusForConditionalGeneration (GraniteSpeechPlusConfig モデル)
- LEDConfig 設定クラス: LEDForConditionalGeneration (LEDConfig モデル)
- LongT5Config 設定クラス: LongT5ForConditionalGeneration (LongT5Config モデル)
- M2M100Config 設定クラス: M2M100ForConditionalGeneration (M2M100Config モデル)
- MBartConfig 設定クラス: MBartForConditionalGeneration (MBartConfig モデル)
- MT5Config 設定クラス: MT5ForConditionalGeneration (MT5Config モデル)
- MarianConfig 設定クラス: MarianMTModel (MarianConfig モデル)
- MossTranscribeDiarizeConfig 設定クラス: MossTranscribeDiarizeForConditionalGeneration (MossTranscribeDiarizeConfig モデル)
- MusicFlamingoConfig 設定クラス: MusicFlamingoForConditionalGeneration (MusicFlamingoConfig モデル)
- MvpConfig 設定クラス: MvpForConditionalGeneration (MvpConfig モデル)
- Nemotron3_5TranscribeConfig 設定クラス: Nemotron3_5TranscribeForConditionalGeneration (Nemotron3_5TranscribeConfig モデル)
- NllbMoeConfig 設定クラス: NllbMoeForConditionalGeneration (NllbMoeConfig モデル)
- PLBartConfig 設定クラス: PLBartForConditionalGeneration (PLBartConfig モデル)
- PegasusConfig 設定クラス: PegasusForConditionalGeneration (PegasusConfig モデル)
- PegasusXConfig 設定クラス: PegasusXForConditionalGeneration (PegasusXConfig モデル)
- ProphetNetConfig 設定クラス: ProphetNetForConditionalGeneration (ProphetNetConfig モデル)
- Qwen2AudioConfig 設定クラス: Qwen2AudioForConditionalGeneration (Qwen2AudioConfig モデル)
- Qwen3ASRConfig 設定クラス: Qwen3ASRForConditionalGeneration (Qwen3ASRConfig モデル)
- SeamlessM4TConfig 設定クラス: SeamlessM4TForTextToText (SeamlessM4TConfig モデル)
- SeamlessM4Tv2Config 設定クラス: SeamlessM4Tv2ForTextToText (SeamlessM4Tv2Config モデル)
- SwitchTransformersConfig 設定クラス: SwitchTransformersForConditionalGeneration (SwitchTransformersConfig モデル)
- T5Config 設定クラス: T5ForConditionalGeneration (T5Config モデル)
- T5Gemma2Config 設定クラス: T5Gemma2ForConditionalGeneration (T5Gemma2Config モデル)
- T5GemmaConfig 設定クラス: T5GemmaForConditionalGeneration (T5GemmaConfig モデル)
- UMT5Config 設定クラス: UMT5ForConditionalGeneration (UMT5Config モデル)
- VibeVoiceAsrConfig 設定クラス: VibeVoiceAsrForConditionalGeneration (VibeVoiceAsrConfig モデル)
- VoxtralConfig 設定クラス: VoxtralForConditionalGeneration (VoxtralConfig モデル)
- VoxtralRealtimeConfig 設定クラス: VoxtralRealtimeForConditionalGeneration (VoxtralRealtimeConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの1つ(シーケンスツーシーケンス言語モデリングヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- audioflamingo3 — AudioFlamingo3ForConditionalGeneration(AudioFlamingo3Config モデル)
- bart — BartForConditionalGeneration(BartConfigモデル)
- bigbird_pegasus — BigBirdPegasusForConditionalGeneration(BigBirdPegasusConfig モデル)
- blenderbot — BlenderbotForConditionalGeneration(BlenderbotConfig モデル)
- blenderbot-small — BlenderbotSmallForConditionalGeneration(BlenderbotSmallConfig モデル)
- エンコーダー・デコーダー — EncoderDecoderModel(EncoderDecoderConfig モデル)
- fsmt — FSMTForConditionalGeneration(FSMTConfig モデル)
- glmasr — GlmAsrForConditionalGeneration(GlmAsrConfig モデル)
- granite_speech — GraniteSpeechForConditionalGeneration(GraniteSpeechConfigモデル)
- granite_speech_plus — GraniteSpeechPlusForConditionalGeneration(GraniteSpeechPlusConfig モデル)
- led — LEDForConditionalGeneration (LEDConfig モデル)
- longt5 — LongT5ForConditionalGeneration(LongT5Config モデル)
- m2m_100 — M2M100ForConditionalGeneration(M2M100Config モデル)
- marian — MarianMTModel(MarianConfig モデル)
- mbart — MBartForConditionalGeneration(MBartConfig モデル)
- moss_transcribe_diarize — MossTranscribeDiarizeForConditionalGeneration(MossTranscribeDiarizeConfig モデル)
- mt5 — MT5ForConditionalGeneration(MT5Config モデル)
- musicflamingo — MusicFlamingoForConditionalGeneration(MusicFlamingoConfig モデル)
- mvp — MvpForConditionalGeneration(MvpConfig モデル)
- nemo_speechlm — Nemotron3_5TranscribeForConditionalGeneration(Nemotron3_5TranscribeConfig モデル)
- nllb-moe — NllbMoeForConditionalGeneration(NllbMoeConfig モデル)
- pegasus — PegasusForConditionalGeneration(PegasusConfig モデル)
- pegasus_x — PegasusXForConditionalGeneration(PegasusXConfigモデル)
- plbart — PLBartForConditionalGeneration(PLBartConfigモデル)
- prophetnet — ProphetNetForConditionalGeneration(ProphetNetConfig モデル)
- qwen2_audio — Qwen2AudioForConditionalGeneration(Qwen2AudioConfig モデル)
- qwen3_asr — Qwen3ASRForConditionalGeneration(Qwen3ASRConfig モデル)
- seamless_m4t — SeamlessM4TForTextToText(SeamlessM4TConfig モデル)
- seamless_m4t_v2 — SeamlessM4Tv2ForTextToText(SeamlessM4Tv2Config モデル)
- switch_transformers — SwitchTransformersForConditionalGeneration(SwitchTransformersConfig モデル)
- t5 — T5ForConditionalGeneration(T5Config モデル)
- t5gemma — T5GemmaForConditionalGeneration(T5GemmaConfig モデル)
- t5gemma2 — T5Gemma2ForConditionalGeneration(T5Gemma2Config モデル)
- umt5 — UMT5ForConditionalGeneration(UMT5Config モデル)
- vibevoice_asr — VibeVoiceAsrForConditionalGeneration(VibeVoiceAsrConfig モデル)
- voxtral — VoxtralForConditionalGeneration(VoxtralConfig モデル)
- voxtral_realtime — VoxtralRealtimeForConditionalGeneration(VoxtralRealtimeConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForSeq2SeqLM
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForSeq2SeqLM.from_pretrained("google-t5/t5-base")
>>> # Update configuration during loading
>>> model = AutoModelForSeq2SeqLM.from_pretrained("google-t5/t5-base", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForSequenceClassification
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスの一つ(シーケンス分類ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- AXK1Config 設定クラス: AXK1ForSequenceClassification (AXK1Config モデル)
- AXK2Config 設定クラス: AXK2ForSequenceClassification (AXK2Config モデル)
- AlbertConfig 設定クラス: AlbertForSequenceClassification (AlbertConfig モデル)
- ArceeConfig 設定クラス: ArceeForSequenceClassification (ArceeConfig モデル)
- BartConfig 設定クラス: BartForSequenceClassification (BartConfig モデル)
- BertConfig 設定クラス: BertForSequenceClassification (BertConfig モデル)
- BigBirdConfig 設定クラス: BigBirdForSequenceClassification (BigBirdConfig モデル)
- BigBirdPegasusConfig 設定クラス: BigBirdPegasusForSequenceClassification (BigBirdPegasusConfig モデル)
- BioGptConfig 設定クラス: BioGptForSequenceClassification (BioGptConfig モデル)
- BloomConfig 設定クラス: BloomForSequenceClassification (BloomConfig モデル)
- CTRLConfig 設定クラス: CTRLForSequenceClassification (CTRLConfig モデル)
- CamembertConfig 設定クラス: CamembertForSequenceClassification (CamembertConfig モデル)
- CanineConfig 設定クラス: CanineForSequenceClassification (CanineConfig モデル)
- CohereCompassTextConfig 設定クラス: CohereCompassTextForSequenceClassification (CohereCompassTextConfig モデル)
- ConvBertConfig 設定クラス: ConvBertForSequenceClassification (ConvBertConfig モデル)
- Data2VecTextConfig 設定クラス: Data2VecTextForSequenceClassification (Data2VecTextConfig モデル)
- DebertaConfig 設定クラス: DebertaForSequenceClassification (DebertaConfig モデル)
- DebertaV2Config 設定クラス: DebertaV2ForSequenceClassification (DebertaV2Config モデル)
- DeepseekV2Config 設定クラス: DeepseekV2ForSequenceClassification (DeepseekV2Config モデル)
- DeepseekV3Config 設定クラス: DeepseekV3ForSequenceClassification (DeepseekV3Config モデル)
- DiffLlamaConfig 設定クラス: DiffLlamaForSequenceClassification (DiffLlamaConfig モデル)
- DistilBertConfig 設定クラス: DistilBertForSequenceClassification (DistilBertConfig モデル)
- DogeConfig 設定クラス: DogeForSequenceClassification (DogeConfig モデル)
- ElectraConfig 設定クラス: ElectraForSequenceClassification (ElectraConfig モデル)
- ErnieConfig 設定クラス: ErnieForSequenceClassification (ErnieConfig モデル)
- EsmConfig 設定クラス: EsmForSequenceClassification (EsmConfig モデル)
- EsmcConfig 設定クラス: EsmcForSequenceClassification (EsmcConfig モデル)
- EuroBertConfig 設定クラス: EuroBertForSequenceClassification (EuroBertConfig モデル)
- Exaone4Config 設定クラス: Exaone4ForSequenceClassification (Exaone4Config モデル)
- FNetConfig 設定クラス: FNetForSequenceClassification (FNetConfig モデル)
- FalconConfig 設定クラス: FalconForSequenceClassification (FalconConfig モデル)
- FlaubertConfig 設定クラス: FlaubertForSequenceClassification (FlaubertConfig モデル)
- FunnelConfig 設定クラス: FunnelForSequenceClassification (FunnelConfig モデル)
- GPT2Config 設定クラス: GPT2ForSequenceClassification (GPT2Config モデル)
- GPTBigCodeConfig 設定クラス: GPTBigCodeForSequenceClassification (GPTBigCodeConfig モデル)
- GPTJConfig 設定クラス: GPTJForSequenceClassification (GPTJConfig モデル)
- GPTNeoConfig 設定クラス: GPTNeoForSequenceClassification (GPTNeoConfig モデル)
- GPTNeoXConfig 設定クラス: GPTNeoXForSequenceClassification (GPTNeoXConfig モデル)
- Gemma2Config 設定クラス: Gemma2ForSequenceClassification (Gemma2Config モデル)
- Gemma3Config 設定クラス: Gemma3ForSequenceClassification (Gemma3Config モデル)
- Gemma3TextConfig 設定クラス: Gemma3TextForSequenceClassification (Gemma3TextConfig モデル)
- GemmaConfig 設定クラス: GemmaForSequenceClassification (GemmaConfig モデル)
- Glm4Config 設定クラス: Glm4ForSequenceClassification (Glm4Config モデル)
- GlmConfig 設定クラス: GlmForSequenceClassification (GlmConfig モデル)
- GptOssConfig 設定クラス: GptOssForSequenceClassification (GptOssConfig モデル)
- GteConfig 設定クラス: GteForSequenceClassification (GteConfig モデル)
- HeliumConfig 設定クラス: HeliumForSequenceClassification (HeliumConfig モデル)
- HunYuanDenseV1Config 設定クラス: HunYuanDenseV1ForSequenceClassification (HunYuanDenseV1Config モデル)
- HunYuanMoEV1Config 設定クラス: HunYuanMoEV1ForSequenceClassification (HunYuanMoEV1Config モデル)
- IBertConfig 設定クラス: IBertForSequenceClassification (IBertConfig モデル)
- JambaConfig 設定クラス: JambaForSequenceClassification (JambaConfig モデル)
- JetMoeConfig 設定クラス: JetMoeForSequenceClassification (JetMoeConfig モデル)
- JinaEmbeddingsV3Config 設定クラス: JinaEmbeddingsV3ForSequenceClassification (JinaEmbeddingsV3Config モデル)
- LayoutLMConfig 設定クラス: LayoutLMForSequenceClassification (LayoutLMConfig モデル)
- LayoutLMv2Config 設定クラス: LayoutLMv2ForSequenceClassification (LayoutLMv2Config モデル)
- LayoutLMv3Config 設定クラス: LayoutLMv3ForSequenceClassification (LayoutLMv3Config モデル)
- LiltConfig 設定クラス: LiltForSequenceClassification (LiltConfig モデル)
- LlamaConfig 設定クラス: LlamaForSequenceClassification (LlamaConfig モデル)
- LongformerConfig 設定クラス: LongformerForSequenceClassification (LongformerConfig モデル)
- LukeConfig 設定クラス: LukeForSequenceClassification (LukeConfig モデル)
- MBartConfig 設定クラス: MBartForSequenceClassification (MBartConfig モデル)
- MPNetConfig 設定クラス: MPNetForSequenceClassification (MPNetConfig モデル)
- MT5Config 設定クラス: MT5ForSequenceClassification (MT5Config モデル)
- MarkupLMConfig 設定クラス: MarkupLMForSequenceClassification (MarkupLMConfig モデル)
- MegatronBertConfig 設定クラス: MegatronBertForSequenceClassification (MegatronBertConfig モデル)
- MiniCPM3Config 設定クラス: MiniCPM3ForSequenceClassification (MiniCPM3Config モデル)
- MiniMaxConfig 設定クラス: MiniMaxForSequenceClassification (MiniMaxConfig モデル)
- Ministral3Config 設定クラス: Ministral3ForSequenceClassification (Ministral3Config モデル)
- MinistralConfig 設定クラス: MinistralForSequenceClassification (MinistralConfig モデル)
- Mistral4Config 設定クラス: Mistral4ForSequenceClassification (Mistral4Config モデル)
- MistralConfig 設定クラス: MistralForSequenceClassification (MistralConfig モデル)
- MixtralConfig 設定クラス: MixtralForSequenceClassification (MixtralConfig モデル)
- MobileBertConfig 設定クラス: MobileBertForSequenceClassification (MobileBertConfig モデル)
- ModernBertConfig 設定クラス: ModernBertForSequenceClassification (ModernBertConfig モデル)
- ModernBertDecoderConfig 設定クラス: ModernBertDecoderForSequenceClassification (ModernBertDecoderConfig モデル)
- ModernVBertConfig 設定クラス: ModernVBertForSequenceClassification (ModernVBertConfig モデル)
- MptConfig 設定クラス: MptForSequenceClassification (MptConfig モデル)
- MraConfig 設定クラス: MraForSequenceClassification (MraConfig モデル)
- MvpConfig 設定クラス: MvpForSequenceClassification (MvpConfig モデル)
- NemotronConfig 設定クラス: NemotronForSequenceClassification (NemotronConfig モデル)
- NomicBertConfig 設定クラス: NomicBertForSequenceClassification (NomicBertConfig モデル)
- NystromformerConfig 設定クラス: NystromformerForSequenceClassification (NystromformerConfig モデル)
- OPTConfig 設定クラス: OPTForSequenceClassification (OPTConfig モデル)
- Olmo2Config 設定クラス: Olmo2ForSequenceClassification (Olmo2Config モデル)
- Olmo3Config 設定クラス: Olmo3ForSequenceClassification (Olmo3Config モデル)
- OlmoConfig 設定クラス: OlmoForSequenceClassification (OlmoConfig モデル)
- OpenAIGPTConfig 設定クラス: OpenAIGPTForSequenceClassification (OpenAIGPTConfig モデル)
- PLBartConfig 設定クラス: PLBartForSequenceClassification (PLBartConfig モデル)
- PerceiverConfig 設定クラス: PerceiverForSequenceClassification (PerceiverConfig モデル)
- PersimmonConfig 設定クラス: PersimmonForSequenceClassification (PersimmonConfig モデル)
- Phi3Config 設定クラス: Phi3ForSequenceClassification (Phi3Config モデル)
- PhiConfig 設定クラス: PhiForSequenceClassification (PhiConfig モデル)
- PhimoeConfig 設定クラス: PhimoeForSequenceClassification (PhimoeConfig モデル)
- Qwen2Config 設定クラス: Qwen2ForSequenceClassification (Qwen2Config モデル)
- Qwen2MoeConfig 設定クラス: Qwen2MoeForSequenceClassification (Qwen2MoeConfig モデル)
- Qwen3Config 設定クラス: Qwen3ForSequenceClassification (Qwen3Config モデル)
- Qwen3MoeConfig 設定クラス: Qwen3MoeForSequenceClassification (Qwen3MoeConfig モデル)
- Qwen3NextConfig 設定クラス: Qwen3NextForSequenceClassification (Qwen3NextConfig モデル)
- Qwen3_5Config 設定クラス: Qwen3_5ForSequenceClassification (Qwen3_5Config モデル)
- Qwen3_5TextConfig 設定クラス: Qwen3_5TextForSequenceClassification (Qwen3_5TextConfig モデル)
- ReformerConfig 設定クラス: ReformerForSequenceClassification (ReformerConfig モデル)
- RemBertConfig 設定クラス: RemBertForSequenceClassification (RemBertConfig モデル)
- RoCBertConfig 設定クラス: RoCBertForSequenceClassification (RoCBertConfig モデル)
- RoFormerConfig 設定クラス: RoFormerForSequenceClassification (RoFormerConfig モデル)
- RobertaConfig 設定クラス: RobertaForSequenceClassification (RobertaConfig モデル)
- RobertaPreLayerNormConfig 設定クラス: RobertaPreLayerNormForSequenceClassification (RobertaPreLayerNormConfig モデル)
- SeedOssConfig 設定クラス: SeedOssForSequenceClassification (SeedOssConfig モデル)
- SmolLM3Config 設定クラス: SmolLM3ForSequenceClassification (SmolLM3Config モデル)
- SqueezeBertConfig 設定クラス: SqueezeBertForSequenceClassification (SqueezeBertConfig モデル)
- StableLmConfig 設定クラス: StableLmForSequenceClassification (StableLmConfig モデル)
- Starcoder2Config 設定クラス: Starcoder2ForSequenceClassification (Starcoder2Config モデル)
- T5Config 設定クラス: T5ForSequenceClassification (T5Config モデル)
- T5Gemma2Config 設定クラス: T5Gemma2ForSequenceClassification (T5Gemma2Config モデル)
- T5GemmaConfig 設定クラス: T5GemmaForSequenceClassification (T5GemmaConfig モデル)
- TapasConfig 設定クラス: TapasForSequenceClassification (TapasConfig モデル)
- UMT5Config 設定クラス: UMT5ForSequenceClassification (UMT5Config モデル)
- UltraVBertConfig 設定クラス: UltraVBertForSequenceClassification (UltraVBertConfig モデル)
- UltrabertConfig 設定クラス: UltrabertForSequenceClassification (UltrabertConfig モデル)
- XLMConfig 設定クラス: XLMForSequenceClassification (XLMConfig モデル)
- XLMRobertaConfig 設定クラス: XLMRobertaForSequenceClassification (XLMRobertaConfig モデル)
- XLMRobertaXLConfig 設定クラス: XLMRobertaXLForSequenceClassification (XLMRobertaXLConfig モデル)
- XLNetConfig 設定クラス: XLNetForSequenceClassification (XLNetConfig モデル)
- XmodConfig 設定クラス: XmodForSequenceClassification (XmodConfig モデル)
- YosoConfig 設定クラス: YosoForSequenceClassification (YosoConfig モデル)
- Zamba2Config 設定クラス: Zamba2ForSequenceClassification (Zamba2Config モデル)
- ZambaConfig 設定クラス: ZambaForSequenceClassification (ZambaConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの1つ(シーケンス分類ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- albert — AlbertForSequenceClassification(AlbertConfig モデル)
- arcee — ArceeForSequenceClassification(ArceeConfigモデル)
- axk1 — AXK1ForSequenceClassification(AXK1Configモデル)
- axk2 — AXK2ForSequenceClassification(AXK2Configモデル)
- bart — BartForSequenceClassification(BartConfig モデル)
- bert — BertForSequenceClassification(BertConfig モデル)
- big_bird — BigBirdForSequenceClassification(BigBirdConfigモデル)
- bigbird_pegasus — BigBirdPegasusForSequenceClassification(BigBirdPegasusConfig モデル)
- biogpt — BioGptForSequenceClassification(BioGptConfig モデル)
- bloom — BloomForSequenceClassification(BloomConfig モデル)
- camembert — CamembertForSequenceClassification(CamembertConfig モデル)
- canine — CanineForSequenceClassification(CanineConfig モデル)
- cohere_compass_text — CohereCompassTextForSequenceClassification(CohereCompassTextConfig モデル)
- convbert — ConvBertForSequenceClassification(ConvBertConfigモデル)
- ctrl — CTRLForSequenceClassification(CTRLConfig モデル)
- data2vec-text — Data2VecTextForSequenceClassification(Data2VecTextConfig モデル)
- deberta — DebertaForSequenceClassification(DebertaConfig モデル)
- deberta-v2 — DebertaV2ForSequenceClassification(DebertaV2Config モデル)
- deepseek_v2 — DeepseekV2ForSequenceClassification(DeepseekV2Config モデル)
- deepseek_v3 — DeepseekV3ForSequenceClassification(DeepseekV3Config モデル)
- diffllama — DiffLlamaForSequenceClassification(DiffLlamaConfigモデル)
- distilbert — DistilBertForSequenceClassification(DistilBertConfig モデル)
- doge — DogeForSequenceClassification(DogeConfigモデル)
- electra — ElectraForSequenceClassification(ElectraConfig モデル)
- ernie — ErnieForSequenceClassification(ErnieConfig モデル)
- esm — EsmForSequenceClassification(EsmConfig モデル)
- esmc — EsmcForSequenceClassification(EsmcConfigモデル)
- eurobert — EuroBertForSequenceClassification(EuroBertConfigモデル)
- exaone4 — Exaone4ForSequenceClassification(Exaone4Configモデル)
- falcon — FalconForSequenceClassification(FalconConfigモデル)
- flaubert — FlaubertForSequenceClassification(FlaubertConfig モデル)
- fnet — FNetForSequenceClassification(FNetConfig モデル)
- funnel — FunnelForSequenceClassification(FunnelConfigモデル)
- gemma — GemmaForSequenceClassification(GemmaConfig モデル)
- gemma2 — Gemma2ForSequenceClassification(Gemma2Config モデル)
- gemma3 — Gemma3ForSequenceClassification(Gemma3Config モデル)
- gemma3_text — Gemma3TextForSequenceClassification(Gemma3TextConfig モデル)
- glm — GlmForSequenceClassification(GlmConfig モデル)
- glm4 — Glm4ForSequenceClassification (Glm4Config モデル)
- gpt-sw3 — GPT2ForSequenceClassification(GPT2Config モデル)
- gpt2 — GPT2ForSequenceClassification(GPT2Config モデル)
- gpt_bigcode — GPTBigCodeForSequenceClassification(GPTBigCodeConfig モデル)
- gpt_neo — GPTNeoForSequenceClassification(GPTNeoConfigモデル)
- gpt_neox — GPTNeoXForSequenceClassification(GPTNeoXConfigモデル)
- gpt_oss — GptOssForSequenceClassification(GptOssConfig モデル)
- gptj — GPTJForSequenceClassification(GPTJConfig モデル)
- gte — GteForSequenceClassification(GteConfig モデル)
- helium — HeliumForSequenceClassification(HeliumConfigモデル)
- hunyuan_v1_dense — HunYuanDenseV1ForSequenceClassification(HunYuanDenseV1Config モデル)
- hunyuan_v1_moe — HunYuanMoEV1ForSequenceClassification(HunYuanMoEV1Config モデル)
- ibert — IBertForSequenceClassification(IBertConfigモデル)
- jamba — JambaForSequenceClassification(JambaConfig モデル)
- jetmoe — JetMoeForSequenceClassification(JetMoeConfigモデル)
- jina_embeddings_v3 — JinaEmbeddingsV3ForSequenceClassification(JinaEmbeddingsV3Config モデル)
- layoutlm — LayoutLMForSequenceClassification (LayoutLMConfig モデル)
- layoutlmv2 — LayoutLMv2ForSequenceClassification(LayoutLMv2Config モデル)
- layoutlmv3 — LayoutLMv3ForSequenceClassification(LayoutLMv3Config モデル)
- lilt — LiltForSequenceClassification(LiltConfig モデル)
- llama — LlamaForSequenceClassification(LlamaConfig モデル)
- longformer — LongformerForSequenceClassification(LongformerConfig モデル)
- luke — LukeForSequenceClassification (LukeConfig モデル)
- markuplm — MarkupLMForSequenceClassification (MarkupLMConfig モデル)
- mbart — MBartForSequenceClassification(MBartConfig モデル)
- megatron-bert — MegatronBertForSequenceClassification(MegatronBertConfig モデル)
- minicpm3 — MiniCPM3ForSequenceClassification(MiniCPM3Config モデル)
- minimax — MiniMaxForSequenceClassification(MiniMaxConfig モデル)
- ministral — MinistralForSequenceClassification(MinistralConfig モデル)
- ministral3 — Ministral3ForSequenceClassification(Ministral3Config モデル)
- mistral — MistralForSequenceClassification(MistralConfig モデル)
- mistral4 — Mistral4ForSequenceClassification(Mistral4Config モデル)
- mixtral — MixtralForSequenceClassification(MixtralConfig モデル)
- mobilebert — MobileBertForSequenceClassification(MobileBertConfig モデル)
- modernbert — ModernBertForSequenceClassification(ModernBertConfigモデル)
- modernbert-decoder — ModernBertDecoderForSequenceClassification(ModernBertDecoderConfig モデル)
- modernvbert — ModernVBertForSequenceClassification(ModernVBertConfig モデル)
- mpnet — MPNetForSequenceClassification(MPNetConfig モデル)
- mpt — MptForSequenceClassification(MptConfig モデル)
- mra — MraForSequenceClassification(MraConfigモデル)
- mt5 — MT5ForSequenceClassification (MT5Config モデル)
- mvp — MvpForSequenceClassification(MvpConfig モデル)
- nemotron — NemotronForSequenceClassification(NemotronConfig モデル)
- nomic_bert — NomicBertForSequenceClassification(NomicBertConfig モデル)
- nystromformer — NystromformerForSequenceClassification(NystromformerConfig モデル)
- olmo — OlmoForSequenceClassification(OlmoConfig モデル)
- olmo2 — Olmo2ForSequenceClassification(Olmo2Config モデル)
- olmo3 — Olmo3ForSequenceClassification(Olmo3Config モデル)
- openai-gpt — OpenAIGPTForSequenceClassification(OpenAIGPTConfig モデル)
- opt — OPTForSequenceClassification(OPTConfig モデル)
- perceiver — PerceiverForSequenceClassification(PerceiverConfigモデル)
- persimmon — PersimmonForSequenceClassification(PersimmonConfig モデル)
- phi — PhiForSequenceClassification (PhiConfig モデル)
- phi3 — Phi3ForSequenceClassification (Phi3Config モデル)
- phimoe — PhimoeForSequenceClassification(PhimoeConfig モデル)
- plbart — PLBartForSequenceClassification(PLBartConfig モデル)
- qwen2 — Qwen2ForSequenceClassification(Qwen2Config モデル)
- qwen2_moe — Qwen2MoeForSequenceClassification(Qwen2MoeConfig モデル)
- qwen3 — Qwen3ForSequenceClassification(Qwen3Config モデル)
- qwen3_5 — Qwen3_5ForSequenceClassification(Qwen3_5Config モデル)
- qwen3_5_text — Qwen3_5TextForSequenceClassification(Qwen3_5TextConfig モデル)
- qwen3_moe — Qwen3MoeForSequenceClassification(Qwen3MoeConfig モデル)
- qwen3_next — Qwen3NextForSequenceClassification(Qwen3NextConfig モデル)
- reformer — ReformerForSequenceClassification(ReformerConfigモデル)
- rembert — RemBertForSequenceClassification(RemBertConfig モデル)
- roberta — RobertaForSequenceClassification(RobertaConfig モデル)
- roberta-prelayernorm — RobertaPreLayerNormForSequenceClassification(RobertaPreLayerNormConfig モデル)
- roc_bert — RoCBertForSequenceClassification(RoCBertConfig モデル)
- roformer — RoFormerForSequenceClassification(RoFormerConfigモデル)
- seed_oss — SeedOssForSequenceClassification(SeedOssConfig モデル)
- smollm3 — SmolLM3ForSequenceClassification(SmolLM3Config モデル)
- squeezebert — SqueezeBertForSequenceClassification(SqueezeBertConfig モデル)
- stablelm — StableLmForSequenceClassification(StableLmConfig モデル)
- starcoder2 — Starcoder2ForSequenceClassification (Starcoder2Config モデル)
- t5 — T5ForSequenceClassification(T5Config モデル)
- t5gemma — T5GemmaForSequenceClassification(T5GemmaConfig モデル)
- t5gemma2 — T5Gemma2ForSequenceClassification(T5Gemma2Config モデル)
- tapas — TapasForSequenceClassification(TapasConfig モデル)
- ultrabert — UltrabertForSequenceClassification(UltrabertConfig モデル)
- ultravbert — UltraVBertForSequenceClassification(UltraVBertConfigモデル)
- umt5 — UMT5ForSequenceClassification(UMT5Config モデル)
- xlm — XLMForSequenceClassification(XLMConfig モデル)
- xlm-roberta — XLMRobertaForSequenceClassification(XLMRobertaConfig モデル)
- xlm-roberta-xl — XLMRobertaXLForSequenceClassification(XLMRobertaXLConfig モデル)
- xlnet — XLNetForSequenceClassification(XLNetConfigモデル)
- xmod — XmodForSequenceClassification(XmodConfigモデル)
- yoso — YosoForSequenceClassification(YosoConfigモデル)
- zamba — ZambaForSequenceClassification(ZambaConfig モデル)
- zamba2 — Zamba2ForSequenceClassification(Zamba2Config モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForSequenceClassification
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForSequenceClassification.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForSequenceClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForMultipleChoice
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスの一つ(マルチチョイスヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- AlbertConfig 設定クラス: AlbertForMultipleChoice (AlbertConfig モデル)
- BertConfig 設定クラス: BertForMultipleChoice (BertConfig モデル)
- BigBirdConfig 設定クラス: BigBirdForMultipleChoice (BigBirdConfig モデル)
- CamembertConfig 設定クラス: CamembertForMultipleChoice (CamembertConfig モデル)
- CanineConfig 設定クラス: CanineForMultipleChoice (CanineConfig モデル)
- ConvBertConfig 設定クラス: ConvBertForMultipleChoice (ConvBertConfig モデル)
- Data2VecTextConfig 設定クラス: Data2VecTextForMultipleChoice (Data2VecTextConfig モデル)
- DebertaV2Config 設定クラス: DebertaV2ForMultipleChoice (DebertaV2Config モデル)
- DistilBertConfig 設定クラス: DistilBertForMultipleChoice (DistilBertConfig モデル)
- ElectraConfig 設定クラス: ElectraForMultipleChoice (ElectraConfig モデル)
- ErnieConfig 設定クラス: ErnieForMultipleChoice (ErnieConfig モデル)
- FNetConfig 設定クラス: FNetForMultipleChoice (FNetConfig モデル)
- FlaubertConfig 設定クラス: FlaubertForMultipleChoice (FlaubertConfig モデル)
- FunnelConfig 設定クラス: FunnelForMultipleChoice (FunnelConfig モデル)
- IBertConfig 設定クラス: IBertForMultipleChoice (IBertConfig モデル)
- LongformerConfig 設定クラス: LongformerForMultipleChoice (LongformerConfig モデル)
- LukeConfig 設定クラス: LukeForMultipleChoice (LukeConfig モデル)
- MPNetConfig 設定クラス: MPNetForMultipleChoice (MPNetConfig モデル)
- MegatronBertConfig 設定クラス: MegatronBertForMultipleChoice (MegatronBertConfig モデル)
- MobileBertConfig 設定クラス: MobileBertForMultipleChoice (MobileBertConfig モデル)
- ModernBertConfig 設定クラス: ModernBertForMultipleChoice (ModernBertConfig モデル)
- MraConfig 設定クラス: MraForMultipleChoice (MraConfig モデル)
- NystromformerConfig 設定クラス: NystromformerForMultipleChoice (NystromformerConfig モデル)
- RemBertConfig 設定クラス: RemBertForMultipleChoice (RemBertConfig モデル)
- RoCBertConfig 設定クラス: RoCBertForMultipleChoice (RoCBertConfig モデル)
- RoFormerConfig 設定クラス: RoFormerForMultipleChoice (RoFormerConfig モデル)
- RobertaConfig 設定クラス: RobertaForMultipleChoice (RobertaConfig モデル)
- RobertaPreLayerNormConfig 設定クラス: RobertaPreLayerNormForMultipleChoice (RobertaPreLayerNormConfig モデル)
- SqueezeBertConfig 設定クラス: SqueezeBertForMultipleChoice (SqueezeBertConfig モデル)
- XLMConfig 設定クラス: XLMForMultipleChoice (XLMConfig モデル)
- XLMRobertaConfig 設定クラス: XLMRobertaForMultipleChoice (XLMRobertaConfig モデル)
- XLMRobertaXLConfig 設定クラス: XLMRobertaXLForMultipleChoice (XLMRobertaXLConfig モデル)
- XLNetConfig 設定クラス: XLNetForMultipleChoice (XLNetConfig モデル)
- XmodConfig 設定クラス: XmodForMultipleChoice (XmodConfig モデル)
- YosoConfig 設定クラス: YosoForMultipleChoice (YosoConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(マルチチョイスヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
ライブラリのモデルクラスの一つ(multiple choice head付き)を、事前学習済みモデルからインスタンス化します。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- albert — AlbertForMultipleChoice(AlbertConfig モデル)
- bert — BertForMultipleChoice(BertConfig モデル)
- big_bird — BigBirdForMultipleChoice(BigBirdConfig モデル)
- camembert — CamembertForMultipleChoice(CamembertConfig モデル)
- canine — CanineForMultipleChoice(CanineConfig モデル)
- convbert — ConvBertForMultipleChoice (ConvBertConfig モデル)
- data2vec-text — Data2VecTextForMultipleChoice (Data2VecTextConfig モデル)
- deberta-v2 — DebertaV2ForMultipleChoice (DebertaV2Config モデル)
- distilbert — DistilBertForMultipleChoice (DistilBertConfig モデル)
- electra — ElectraForMultipleChoice(ElectraConfig モデル)
- ernie — ErnieForMultipleChoice(ErnieConfigモデル)
- flaubert — FlaubertForMultipleChoice(FlaubertConfig モデル)
- fnet — FNetForMultipleChoice (FNetConfig モデル)
- funnel — FunnelForMultipleChoice(FunnelConfigモデル)
- ibert — IBertForMultipleChoice(IBertConfigモデル)
- longformer — LongformerForMultipleChoice(LongformerConfig モデル)
- luke — LukeForMultipleChoice(LukeConfig モデル)
- megatron-bert — MegatronBertForMultipleChoice(MegatronBertConfig モデル)
- mobilebert — MobileBertForMultipleChoice(MobileBertConfig モデル)
- modernbert — ModernBertForMultipleChoice (ModernBertConfig モデル)
- mpnet — MPNetForMultipleChoice(MPNetConfig モデル)
- mra — MraForMultipleChoice(MraConfigモデル)
- nystromformer — NystromformerForMultipleChoice(NystromformerConfig モデル)
- rembert — RemBertForMultipleChoice (RemBertConfig モデル)
- roberta — RobertaForMultipleChoice(RobertaConfigモデル)
- roberta-prelayernorm — RobertaPreLayerNormForMultipleChoice(RobertaPreLayerNormConfig モデル)
- roc_bert — RoCBertForMultipleChoice(RoCBertConfig モデル)
- roformer — RoFormerForMultipleChoice(RoFormerConfig モデル)
- squeezebert — SqueezeBertForMultipleChoice(SqueezeBertConfigモデル)
- xlm — XLMForMultipleChoice (XLMConfig モデル)
- xlm-roberta — XLMRobertaForMultipleChoice(XLMRobertaConfig モデル)
- xlm-roberta-xl — XLMRobertaXLForMultipleChoice(XLMRobertaXLConfig モデル)
- xlnet — XLNetForMultipleChoice(XLNetConfigモデル)
- xmod — XmodForMultipleChoice(XmodConfig モデル)
- yoso — YosoForMultipleChoice(YosoConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForMultipleChoice
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForMultipleChoice.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForMultipleChoice.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForNextSentencePrediction
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスの一つ(次の文予測ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- BertConfig 設定クラス: BertForNextSentencePrediction (BertConfig モデル)
- ErnieConfig 設定クラス: ErnieForNextSentencePrediction (ErnieConfig モデル)
- FNetConfig 設定クラス: FNetForNextSentencePrediction (FNetConfig モデル)
- MegatronBertConfig 設定クラス: MegatronBertForNextSentencePrediction (MegatronBertConfig モデル)
- MobileBertConfig 設定クラス: MobileBertForNextSentencePrediction (MobileBertConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(次の文の予測ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- bert — BertForNextSentencePrediction (BertConfig モデル)
- ernie — ErnieForNextSentencePrediction(ErnieConfig モデル)
- fnet — FNetForNextSentencePrediction(FNetConfig モデル)
- megatron-bert — MegatronBertForNextSentencePrediction(MegatronBertConfig モデル)
- mobilebert — MobileBertForNextSentencePrediction(MobileBertConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForNextSentencePrediction
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForNextSentencePrediction.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForNextSentencePrediction.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForTokenClassification
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、トークン分類ヘッドを持つライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- AXK1Config 設定クラス: AXK1ForTokenClassification (AXK1Config モデル)
- AXK2Config 設定クラス: AXK2ForTokenClassification (AXK2Config モデル)
- AlbertConfig 設定クラス: AlbertForTokenClassification (AlbertConfig モデル)
- ApertusConfig 設定クラス: ApertusForTokenClassification (ApertusConfig モデル)
- ArceeConfig 設定クラス: ArceeForTokenClassification (ArceeConfig モデル)
- BertConfig 設定クラス: BertForTokenClassification (BertConfig モデル)
- BigBirdConfig 設定クラス: BigBirdForTokenClassification (BigBirdConfig モデル)
- BioGptConfig 設定クラス: BioGptForTokenClassification (BioGptConfig モデル)
- BloomConfig 設定クラス: BloomForTokenClassification (BloomConfig モデル)
- BrosConfig 設定クラス: BrosForTokenClassification (BrosConfig モデル)
- CamembertConfig 設定クラス: CamembertForTokenClassification (CamembertConfig モデル)
- CanineConfig 設定クラス: CanineForTokenClassification (CanineConfig モデル)
- ConvBertConfig 設定クラス: ConvBertForTokenClassification (ConvBertConfig モデル)
- Data2VecTextConfig 設定クラス: Data2VecTextForTokenClassification (Data2VecTextConfig モデル)
- DebertaConfig 設定クラス: DebertaForTokenClassification (DebertaConfig モデル)
- DebertaV2Config 設定クラス: DebertaV2ForTokenClassification (DebertaV2Config モデル)
- DeepseekV3Config 設定クラス: DeepseekV3ForTokenClassification (DeepseekV3Config モデル)
- DiffLlamaConfig 設定クラス: DiffLlamaForTokenClassification (DiffLlamaConfig モデル)
- DistilBertConfig 設定クラス: DistilBertForTokenClassification (DistilBertConfig モデル)
- ElectraConfig 設定クラス: ElectraForTokenClassification (ElectraConfig モデル)
- ErnieConfig 設定クラス: ErnieForTokenClassification (ErnieConfig モデル)
- EsmConfig 設定クラス: EsmForTokenClassification (EsmConfig モデル)
- EsmcConfig 設定クラス: EsmcForTokenClassification (EsmcConfig モデル)
- EuroBertConfig 設定クラス: EuroBertForTokenClassification (EuroBertConfig モデル)
- Exaone4Config 設定クラス: Exaone4ForTokenClassification (Exaone4Config モデル)
- FNetConfig 設定クラス: FNetForTokenClassification (FNetConfig モデル)
- FalconConfig 設定クラス: FalconForTokenClassification (FalconConfig モデル)
- FlaubertConfig 設定クラス: FlaubertForTokenClassification (FlaubertConfig モデル)
- FunnelConfig 設定クラス: FunnelForTokenClassification (FunnelConfig モデル)
- GPT2Config 設定クラス: GPT2ForTokenClassification (GPT2Config モデル)
- GPTBigCodeConfig 設定クラス: GPTBigCodeForTokenClassification (GPTBigCodeConfig モデル)
- GPTNeoConfig 設定クラス: GPTNeoForTokenClassification (GPTNeoConfig モデル)
- GPTNeoXConfig 設定クラス: GPTNeoXForTokenClassification (GPTNeoXConfig モデル)
- Gemma2Config 設定クラス: Gemma2ForTokenClassification (Gemma2Config モデル)
- GemmaConfig 設定クラス: GemmaForTokenClassification (GemmaConfig モデル)
- Glm4Config 設定クラス: Glm4ForTokenClassification (Glm4Config モデル)
- GlmConfig 設定クラス: GlmForTokenClassification (GlmConfig モデル)
- GptOssConfig 設定クラス: GptOssForTokenClassification (GptOssConfig モデル)
- GteConfig 設定クラス: GteForTokenClassification (GteConfig モデル)
- HeliumConfig 設定クラス: HeliumForTokenClassification (HeliumConfig モデル)
- IBertConfig 設定クラス: IBertForTokenClassification (IBertConfig モデル)
- JinaEmbeddingsV3Config 設定クラス: JinaEmbeddingsV3ForTokenClassification (JinaEmbeddingsV3Config モデル)
- LayoutLMConfig 設定クラス: LayoutLMForTokenClassification (LayoutLMConfig モデル)
- LayoutLMv2Config 設定クラス: LayoutLMv2ForTokenClassification (LayoutLMv2Config モデル)
- LayoutLMv3Config 設定クラス: LayoutLMv3ForTokenClassification (LayoutLMv3Config モデル)
- LiltConfig 設定クラス: LiltForTokenClassification (LiltConfig モデル)
- LlamaConfig 設定クラス: LlamaForTokenClassification (LlamaConfig モデル)
- LongformerConfig 設定クラス: LongformerForTokenClassification (LongformerConfig モデル)
- LukeConfig 設定クラス: LukeForTokenClassification (LukeConfig モデル)
- MPNetConfig 設定クラス: MPNetForTokenClassification (MPNetConfig モデル)
- MT5Config 設定クラス: MT5ForTokenClassification (MT5Config モデル)
- MarkupLMConfig 設定クラス: MarkupLMForTokenClassification (MarkupLMConfig モデル)
- MegatronBertConfig 設定クラス: MegatronBertForTokenClassification (MegatronBertConfig モデル)
- MiniMaxConfig 設定クラス: MiniMaxForTokenClassification (MiniMaxConfig モデル)
- Ministral3Config 設定クラス: Ministral3ForTokenClassification (Ministral3Config モデル)
- MinistralConfig 設定クラス: MinistralForTokenClassification (MinistralConfig モデル)
- Mistral4Config 設定クラス: Mistral4ForTokenClassification(Mistral4Config モデル)
- MistralConfig 設定クラス: MistralForTokenClassification (MistralConfig モデル)
- MixtralConfig 設定クラス: MixtralForTokenClassification (MixtralConfig モデル)
- MobileBertConfig 設定クラス: MobileBertForTokenClassification (MobileBertConfig モデル)
- ModernBertConfig 設定クラス: ModernBertForTokenClassification (ModernBertConfig モデル)
- ModernVBertConfig 設定クラス: ModernVBertForTokenClassification (ModernVBertConfig モデル)
- MptConfig 設定クラス: MptForTokenClassification (MptConfig モデル)
- MraConfig 設定クラス: MraForTokenClassification (MraConfig モデル)
- NemotronConfig 設定クラス: NemotronForTokenClassification (NemotronConfig モデル)
- NomicBertConfig 設定クラス: NomicBertForTokenClassification (NomicBertConfig モデル)
- NystromformerConfig 設定クラス: NystromformerForTokenClassification (NystromformerConfig モデル)
- OpenAIPrivacyFilterConfig 設定クラス: OpenAIPrivacyFilterForTokenClassification (OpenAIPrivacyFilterConfig モデル)
- PersimmonConfig 設定クラス: PersimmonForTokenClassification (PersimmonConfig モデル)
- Phi3Config 設定クラス: Phi3ForTokenClassification (Phi3Config モデル)
- PhiConfig 設定クラス: PhiForTokenClassification (PhiConfig モデル)
- Qwen2Config 設定クラス: Qwen2ForTokenClassification (Qwen2Config モデル)
- Qwen2MoeConfig 設定クラス: Qwen2MoeForTokenClassification (Qwen2MoeConfig モデル)
- Qwen3ASRConfig 設定クラス: Qwen3ASRForTokenClassification (Qwen3ASRConfig モデル)
- Qwen3Config 設定クラス: Qwen3ForTokenClassification (Qwen3Config モデル)
- Qwen3MoeConfig 設定クラス: Qwen3MoeForTokenClassification (Qwen3MoeConfig モデル)
- Qwen3NextConfig 設定クラス: Qwen3NextForTokenClassification (Qwen3NextConfig モデル)
- Qwen3_5Config 設定クラス: Qwen3_5ForTokenClassification (Qwen3_5Config モデル)
- RemBertConfig 設定クラス: RemBertForTokenClassification (RemBertConfig モデル)
- RoCBertConfig 設定クラス: RoCBertForTokenClassification (RoCBertConfig モデル)
- RoFormerConfig 設定クラス: RoFormerForTokenClassification (RoFormerConfig モデル)
- RobertaConfig 設定クラス: RobertaForTokenClassification (RobertaConfig モデル)
- RobertaPreLayerNormConfig 設定クラス: RobertaPreLayerNormForTokenClassification (RobertaPreLayerNormConfig モデル)
- SeedOssConfig 設定クラス: SeedOssForTokenClassification (SeedOssConfig モデル)
- SmolLM3Config 設定クラス: SmolLM3ForTokenClassification (SmolLM3Config モデル)
- SqueezeBertConfig 設定クラス: SqueezeBertForTokenClassification (SqueezeBertConfig モデル)
- StableLmConfig 設定クラス: StableLmForTokenClassification (StableLmConfig モデル)
- Starcoder2Config 設定クラス: Starcoder2ForTokenClassification (Starcoder2Config モデル)
- T5Config 設定クラス: T5ForTokenClassification (T5Config モデル)
- T5Gemma2Config 設定クラス: T5Gemma2ForTokenClassification (T5Gemma2Config モデル)
- T5GemmaConfig 設定クラス: T5GemmaForTokenClassification (T5GemmaConfig モデル)
- UMT5Config 設定クラス: UMT5ForTokenClassification (UMT5Config モデル)
- UltraVBertConfig 設定クラス: UltraVBertForTokenClassification (UltraVBertConfig モデル)
- UltrabertConfig 設定クラス: UltrabertForTokenClassification (UltrabertConfig モデル)
- XLMConfig 設定クラス: XLMForTokenClassification (XLMConfig モデル)
- XLMRobertaConfig 設定クラス: XLMRobertaForTokenClassification (XLMRobertaConfig モデル)
- XLMRobertaXLConfig 設定クラス: XLMRobertaXLForTokenClassification (XLMRobertaXLConfig モデル)
- XLNetConfig 設定クラス: XLNetForTokenClassification (XLNetConfig モデル)
- XmodConfig 設定クラス: XmodForTokenClassification (XmodConfig モデル)
- YosoConfig 設定クラス: YosoForTokenClassification (YosoConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(トークン分類ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- albert — AlbertForTokenClassification(AlbertConfig モデル)
- apertus — ApertusForTokenClassification(ApertusConfigモデル)
- arcee — ArceeForTokenClassification(ArceeConfigモデル)
- axk1 — AXK1ForTokenClassification(AXK1Configモデル)
- axk2 — AXK2ForTokenClassification(AXK2Config モデル)
- bert — BertForTokenClassification(BertConfig モデル)
- big_bird — BigBirdForTokenClassification(BigBirdConfigモデル)
- biogpt — BioGptForTokenClassification(BioGptConfig モデル)
- bloom — BloomForTokenClassification (BloomConfig モデル)
- bros — BrosForTokenClassification (BrosConfig モデル)
- camembert — CamembertForTokenClassification(CamembertConfig モデル)
- canine — CanineForTokenClassification(CanineConfig モデル)
- convbert — ConvBertForTokenClassification(ConvBertConfig モデル)
- data2vec-text — Data2VecTextForTokenClassification(Data2VecTextConfig モデル)
- deberta — DebertaForTokenClassification(DebertaConfig モデル)
- deberta-v2 — DebertaV2ForTokenClassification(DebertaV2Config モデル)
- deepseek_v3 — DeepseekV3ForTokenClassification(DeepseekV3Config モデル)
- diffllama — DiffLlamaForTokenClassification(DiffLlamaConfig モデル)
- distilbert — DistilBertForTokenClassification(DistilBertConfigモデル)
- electra — ElectraForTokenClassification(ElectraConfig モデル)
- ernie — ErnieForTokenClassification(ErnieConfig モデル)
- esm — EsmForTokenClassification(EsmConfigモデル)
- esmc — EsmcForTokenClassification(EsmcConfigモデル)
- eurobert — EuroBertForTokenClassification(EuroBertConfigモデル)
- exaone4 — Exaone4ForTokenClassification(Exaone4Config モデル)
- falcon — FalconForTokenClassification(FalconConfig モデル)
- flaubert — FlaubertForTokenClassification(FlaubertConfig モデル)
- fnet — FNetForTokenClassification (FNetConfig モデル)
- funnel — FunnelForTokenClassification(FunnelConfigモデル)
- gemma — GemmaForTokenClassification(GemmaConfig モデル)
- gemma2 — Gemma2ForTokenClassification(Gemma2Config モデル)
- glm — GlmForTokenClassification(GlmConfig モデル)
- glm4 — Glm4ForTokenClassification(Glm4Config モデル)
- gpt-sw3 — GPT2ForTokenClassification(GPT2Config モデル)
- gpt2 — GPT2ForTokenClassification (GPT2Config モデル)
- gpt_bigcode — GPTBigCodeForTokenClassification(GPTBigCodeConfigモデル)
- gpt_neo — GPTNeoForTokenClassification(GPTNeoConfig モデル)
- gpt_neox — GPTNeoXForTokenClassification(GPTNeoXConfig モデル)
- gpt_oss — GptOssForTokenClassification(GptOssConfigモデル)
- gte — GteForTokenClassification(GteConfig モデル)
- helium — HeliumForTokenClassification(HeliumConfigモデル)
- ibert — IBertForTokenClassification(IBertConfigモデル)
- jina_embeddings_v3 — JinaEmbeddingsV3ForTokenClassification(JinaEmbeddingsV3Config モデル)
- layoutlm — LayoutLMForTokenClassification(LayoutLMConfig モデル)
- layoutlmv2 — LayoutLMv2ForTokenClassification(LayoutLMv2Config モデル)
- layoutlmv3 — LayoutLMv3ForTokenClassification(LayoutLMv3Config モデル)
- lilt — LiltForTokenClassification(LiltConfigモデル)
- llama — LlamaForTokenClassification(LlamaConfigモデル)
- longformer — LongformerForTokenClassification(LongformerConfig モデル)
- luke — LukeForTokenClassification(LukeConfig モデル)
- markuplm — MarkupLMForTokenClassification(MarkupLMConfig モデル)
- megatron-bert — MegatronBertForTokenClassification (MegatronBertConfig モデル)
- minimax — MiniMaxForTokenClassification(MiniMaxConfigモデル)
- ministral — MinistralForTokenClassification(MinistralConfig モデル)
- ministral3 — Ministral3ForTokenClassification(Ministral3Config モデル)
- mistral — MistralForTokenClassification(MistralConfig モデル)
- mistral4 — Mistral4ForTokenClassification(Mistral4Config モデル)
- mixtral — MixtralForTokenClassification(MixtralConfig モデル)
- mobilebert — MobileBertForTokenClassification(MobileBertConfig モデル)
- modernbert — ModernBertForTokenClassification(ModernBertConfig モデル)
- modernvbert — ModernVBertForTokenClassification(ModernVBertConfigモデル)
- mpnet — MPNetForTokenClassification(MPNetConfig モデル)
- mpt — MptForTokenClassification (MptConfig モデル)
- mra — MraForTokenClassification(MraConfigモデル)
- mt5 — MT5ForTokenClassification(MT5Config モデル)
- nemotron — NemotronForTokenClassification(NemotronConfig モデル)
- nomic_bert — NomicBertForTokenClassification(NomicBertConfigモデル)
- nystromformer — NystromformerForTokenClassification(NystromformerConfig モデル)
- openai_privacy_filter — OpenAIPrivacyFilterForTokenClassification(OpenAIPrivacyFilterConfig モデル)
- persimmon — PersimmonForTokenClassification(PersimmonConfig モデル)
- phi — PhiForTokenClassification(PhiConfig モデル)
- phi3 — Phi3ForTokenClassification(Phi3Config モデル)
- qwen2 — Qwen2ForTokenClassification(Qwen2Config モデル)
- qwen2_moe — Qwen2MoeForTokenClassification(Qwen2MoeConfig モデル)
- qwen3 — Qwen3ForTokenClassification(Qwen3Config モデル)
- qwen3_5 — Qwen3_5ForTokenClassification(Qwen3_5Config モデル)
- qwen3_asr — Qwen3ASRForTokenClassification(Qwen3ASRConfig モデル)
- qwen3_moe — Qwen3MoeForTokenClassification(Qwen3MoeConfig モデル)
- qwen3_next — Qwen3NextForTokenClassification(Qwen3NextConfigモデル)
- rembert — RemBertForTokenClassification(RemBertConfig モデル)
- roberta — RobertaForTokenClassification(RobertaConfig モデル)
- roberta-prelayernorm — RobertaPreLayerNormForTokenClassification(RobertaPreLayerNormConfigモデル)
- roc_bert — RoCBertForTokenClassification(RoCBertConfig モデル)
- roformer — RoFormerForTokenClassification(RoFormerConfig モデル)
- seed_oss — SeedOssForTokenClassification(SeedOssConfig モデル)
- smollm3 — SmolLM3ForTokenClassification(SmolLM3Config モデル)
- squeezebert — SqueezeBertForTokenClassification(SqueezeBertConfig モデル)
- stablelm — StableLmForTokenClassification(StableLmConfigモデル)
- starcoder2 — Starcoder2ForTokenClassification (Starcoder2Config モデル)
- t5 — T5ForTokenClassification (T5Config モデル)
- t5gemma — T5GemmaForTokenClassification(T5GemmaConfig モデル)
- t5gemma2 — T5Gemma2ForTokenClassification (T5Gemma2Config モデル)
- ultrabert — UltrabertForTokenClassification(UltrabertConfig モデル)
- ultravbert — UltraVBertForTokenClassification (UltraVBertConfig モデル)
- umt5 — UMT5ForTokenClassification (UMT5Config モデル)
- xlm — XLMForTokenClassification (XLMConfig モデル)
- xlm-roberta — XLMRobertaForTokenClassification(XLMRobertaConfig モデル)
- xlm-roberta-xl — XLMRobertaXLForTokenClassification(XLMRobertaXLConfig モデル)
- xlnet — XLNetForTokenClassification(XLNetConfig モデル)
- xmod — XmodForTokenClassification(XmodConfig モデル)
- yoso — YosoForTokenClassification(YosoConfigモデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForTokenClassification
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForTokenClassification.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForTokenClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForQuestionAnswering
これは汎用的なモデルクラスであり、from_pretrained() クラスメソッドまたは from_config() クラスメソッドで生成された場合、質問応答ヘッドを備えたライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- AlbertConfig 設定クラス: AlbertForQuestionAnswering (AlbertConfig モデル)
- ArceeConfig 設定クラス: ArceeForQuestionAnswering (ArceeConfig モデル)
- BartConfig 設定クラス: BartForQuestionAnswering (BartConfig モデル)
- BertConfig 設定クラス: BertForQuestionAnswering (BertConfig モデル)
- BigBirdConfig 設定クラス: BigBirdForQuestionAnswering (BigBirdConfig モデル)
- BigBirdPegasusConfig 設定クラス: BigBirdPegasusForQuestionAnswering (BigBirdPegasusConfig モデル)
- BloomConfig 設定クラス: BloomForQuestionAnswering (BloomConfig モデル)
- CamembertConfig 設定クラス: CamembertForQuestionAnswering (CamembertConfig モデル)
- CanineConfig 設定クラス: CanineForQuestionAnswering (CanineConfig モデル)
- ConvBertConfig 設定クラス: ConvBertForQuestionAnswering (ConvBertConfig モデル)
- Data2VecTextConfig 設定クラス: Data2VecTextForQuestionAnswering (Data2VecTextConfig モデル)
- DebertaConfig 設定クラス: DebertaForQuestionAnswering (DebertaConfig モデル)
- DebertaV2Config 設定クラス: DebertaV2ForQuestionAnswering (DebertaV2Config モデル)
- DiffLlamaConfig 設定クラス: DiffLlamaForQuestionAnswering (DiffLlamaConfig モデル)
- DistilBertConfig 設定クラス: DistilBertForQuestionAnswering (DistilBertConfig モデル)
- ElectraConfig 設定クラス: ElectraForQuestionAnswering (ElectraConfig モデル)
- ErnieConfig 設定クラス: ErnieForQuestionAnswering (ErnieConfig モデル)
- Exaone4Config 設定クラス: Exaone4ForQuestionAnswering (Exaone4Config モデル)
- FNetConfig 設定クラス: FNetForQuestionAnswering (FNetConfig モデル)
- FalconConfig 設定クラス: FalconForQuestionAnswering (FalconConfig モデル)
- FlaubertConfig 設定クラス: FlaubertForQuestionAnsweringSimple (FlaubertConfig モデル)
- FunnelConfig 設定クラス: FunnelForQuestionAnswering (FunnelConfig モデル)
- GPT2Config 設定クラス: GPT2ForQuestionAnswering (GPT2Config モデル)
- GPTJConfig 設定クラス: GPTJForQuestionAnswering (GPTJConfig モデル)
- GPTNeoConfig 設定クラス: GPTNeoForQuestionAnswering (GPTNeoConfig モデル)
- GPTNeoXConfig 設定クラス: GPTNeoXForQuestionAnswering (GPTNeoXConfig モデル)
- IBertConfig 設定クラス: IBertForQuestionAnswering (IBertConfig モデル)
- JinaEmbeddingsV3Config 設定クラス: JinaEmbeddingsV3ForQuestionAnswering (JinaEmbeddingsV3Config モデル)
- LEDConfig 設定クラス: LEDForQuestionAnswering (LEDConfig モデル)
- LayoutLMv2Config 設定クラス: LayoutLMv2ForQuestionAnswering (LayoutLMv2Config モデル)
- LayoutLMv3Config 設定クラス: LayoutLMv3ForQuestionAnswering (LayoutLMv3Config モデル)
- LiltConfig 設定クラス: LiltForQuestionAnswering (LiltConfig モデル)
- LlamaConfig 設定クラス: LlamaForQuestionAnswering (LlamaConfig モデル)
- LongformerConfig 設定クラス: LongformerForQuestionAnswering (LongformerConfig モデル)
- LukeConfig 設定クラス: LukeForQuestionAnswering (LukeConfig モデル)
- LxmertConfig 設定クラス: LxmertForQuestionAnswering (LxmertConfig モデル)
- MBartConfig 設定クラス: MBartForQuestionAnswering (MBartConfig モデル)
- MPNetConfig 設定クラス: MPNetForQuestionAnswering (MPNetConfig モデル)
- MT5Config 設定クラス: MT5ForQuestionAnswering (MT5Config モデル)
- MarkupLMConfig 設定クラス: MarkupLMForQuestionAnswering (MarkupLMConfig モデル)
- MegatronBertConfig 設定クラス: MegatronBertForQuestionAnswering (MegatronBertConfig モデル)
- MiniMaxConfig 設定クラス: MiniMaxForQuestionAnswering (MiniMaxConfig モデル)
- Ministral3Config 設定クラス: Ministral3ForQuestionAnswering (Ministral3Config モデル)
- MinistralConfig 設定クラス: MinistralForQuestionAnswering (MinistralConfig モデル)
- MistralConfig 設定クラス: MistralForQuestionAnswering (MistralConfig モデル)
- MixtralConfig 設定クラス: MixtralForQuestionAnswering (MixtralConfig モデル)
- MobileBertConfig 設定クラス: MobileBertForQuestionAnswering (MobileBertConfig モデル)
- ModernBertConfig 設定クラス: ModernBertForQuestionAnswering (ModernBertConfig モデル)
- MptConfig 設定クラス: MptForQuestionAnswering (MptConfig モデル)
- MraConfig 設定クラス: MraForQuestionAnswering (MraConfig モデル)
- MvpConfig 設定クラス: MvpForQuestionAnswering (MvpConfig モデル)
- NemotronConfig 設定クラス: NemotronForQuestionAnswering (NemotronConfig モデル)
- NystromformerConfig 設定クラス: NystromformerForQuestionAnswering (NystromformerConfig モデル)
- OPTConfig 設定クラス: OPTForQuestionAnswering (OPTConfig モデル)
- Qwen2Config 設定クラス: Qwen2ForQuestionAnswering (Qwen2Config モデル)
- Qwen2MoeConfig 設定クラス: Qwen2MoeForQuestionAnswering (Qwen2MoeConfig モデル)
- Qwen3Config 設定クラス: Qwen3ForQuestionAnswering (Qwen3Config モデル)
- Qwen3MoeConfig 設定クラス: Qwen3MoeForQuestionAnswering (Qwen3MoeConfig モデル)
- Qwen3NextConfig 設定クラス: Qwen3NextForQuestionAnswering (Qwen3NextConfig モデル)
- ReformerConfig 設定クラス: ReformerForQuestionAnswering (ReformerConfig モデル)
- RemBertConfig 設定クラス: RemBertForQuestionAnswering (RemBertConfig モデル)
- RoCBertConfig 設定クラス: RoCBertForQuestionAnswering (RoCBertConfig モデル)
- RoFormerConfig 設定クラス: RoFormerForQuestionAnswering (RoFormerConfig モデル)
- RobertaConfig 設定クラス: RobertaForQuestionAnswering (RobertaConfig モデル)
- RobertaPreLayerNormConfig 設定クラス: RobertaPreLayerNormForQuestionAnswering (RobertaPreLayerNormConfig モデル)
- SeedOssConfig 設定クラス: SeedOssForQuestionAnswering (SeedOssConfig モデル)
- SmolLM3Config 設定クラス: SmolLM3ForQuestionAnswering (SmolLM3Config モデル)
- SplinterConfig 設定クラス: SplinterForQuestionAnswering (SplinterConfig モデル)
- SqueezeBertConfig 設定クラス: SqueezeBertForQuestionAnswering (SqueezeBertConfig モデル)
- T5Config 設定クラス: T5ForQuestionAnswering (T5Config モデル)
- UMT5Config 設定クラス: UMT5ForQuestionAnswering (UMT5Config モデル)
- UltrabertConfig 設定クラス: UltrabertForQuestionAnswering (UltrabertConfig モデル)
- XLMConfig 設定クラス: XLMForQuestionAnsweringSimple (XLMConfig モデル)
- XLMRobertaConfig 設定クラス: XLMRobertaForQuestionAnswering (XLMRobertaConfig モデル)
- XLMRobertaXLConfig 設定クラス: XLMRobertaXLForQuestionAnswering (XLMRobertaXLConfig モデル)
- XLNetConfig 設定クラス: XLNetForQuestionAnsweringSimple (XLNetConfig モデル)
- XmodConfig 設定クラス: XmodForQuestionAnswering (XmodConfig モデル)
- YosoConfig 設定クラス: YosoForQuestionAnswering (YosoConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(質問応答ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- albert — AlbertForQuestionAnswering(AlbertConfigモデル)
- arcee — ArceeForQuestionAnswering(ArceeConfigモデル)
- bart — BartForQuestionAnswering(BartConfig モデル)
- bert — BertForQuestionAnswering(BertConfig モデル)
- big_bird — BigBirdForQuestionAnswering(BigBirdConfig モデル)
- bigbird_pegasus — BigBirdPegasusForQuestionAnswering(BigBirdPegasusConfig モデル)
- bloom — BloomForQuestionAnswering(BloomConfig モデル)
- camembert — CamembertForQuestionAnswering(CamembertConfig モデル)
- canine — CanineForQuestionAnswering(CanineConfigモデル)
- convbert — ConvBertForQuestionAnswering(ConvBertConfig モデル)
- data2vec-text — Data2VecTextForQuestionAnswering(Data2VecTextConfig モデル)
- deberta — DebertaForQuestionAnswering(DebertaConfig モデル)
- deberta-v2 — DebertaV2ForQuestionAnswering(DebertaV2Config モデル)
- diffllama — DiffLlamaForQuestionAnswering(DiffLlamaConfig モデル)
- distilbert — DistilBertForQuestionAnswering(DistilBertConfigモデル)
- electra — ElectraForQuestionAnswering(ElectraConfig モデル)
- ernie — ErnieForQuestionAnswering(ErnieConfigモデル)
- exaone4 — Exaone4ForQuestionAnswering(Exaone4Config モデル)
- falcon — FalconForQuestionAnswering(FalconConfig モデル)
- flaubert — FlaubertForQuestionAnsweringSimple(FlaubertConfig モデル)
- fnet — FNetForQuestionAnswering(FNetConfig モデル)
- funnel — FunnelForQuestionAnswering(FunnelConfigモデル)
- gpt2 — GPT2ForQuestionAnswering (GPT2Config モデル)
- gpt_neo — GPTNeoForQuestionAnswering(GPTNeoConfigモデル)
- gpt_neox — GPTNeoXForQuestionAnswering(GPTNeoXConfig モデル)
- gptj — GPTJForQuestionAnswering (GPTJConfig モデル)
- ibert — IBertForQuestionAnswering(IBertConfig モデル)
- jina_embeddings_v3 — JinaEmbeddingsV3ForQuestionAnswering(JinaEmbeddingsV3Config モデル)
- layoutlmv2 — LayoutLMv2ForQuestionAnswering(LayoutLMv2Config モデル)
- layoutlmv3 — LayoutLMv3ForQuestionAnswering(LayoutLMv3Config モデル)
- led — LEDForQuestionAnswering(LEDConfig モデル)
- lilt — LiltForQuestionAnswering(LiltConfigモデル)
- llama — LlamaForQuestionAnswering(LlamaConfig モデル)
- longformer — LongformerForQuestionAnswering(LongformerConfig モデル)
- luke — LukeForQuestionAnswering(LukeConfig モデル)
- lxmert — LxmertForQuestionAnswering(LxmertConfig モデル)
- markuplm — MarkupLMForQuestionAnswering(MarkupLMConfig モデル)
- mbart — MBartForQuestionAnswering(MBartConfig モデル)
- megatron-bert — MegatronBertForQuestionAnswering(MegatronBertConfig モデル)
- minimax — MiniMaxForQuestionAnswering(MiniMaxConfig モデル)
- ministral — MinistralForQuestionAnswering(MinistralConfig モデル)
- ministral3 — Ministral3ForQuestionAnswering(Ministral3Config モデル)
- mistral — MistralForQuestionAnswering(MistralConfig モデル)
- mixtral — MixtralForQuestionAnswering(MixtralConfig モデル)
- mobilebert — MobileBertForQuestionAnswering(MobileBertConfig モデル)
- modernbert — ModernBertForQuestionAnswering(ModernBertConfig モデル)
- mpnet — MPNetForQuestionAnswering(MPNetConfig モデル)
- mpt — MptForQuestionAnswering(MptConfig モデル)
- mra — MraForQuestionAnswering(MraConfigモデル)
- mt5 — MT5ForQuestionAnswering(MT5Config モデル)
- mvp — MvpForQuestionAnswering(MvpConfig モデル)
- nemotron — NemotronForQuestionAnswering(NemotronConfig モデル)
- nystromformer — NystromformerForQuestionAnswering(NystromformerConfig モデル)
- opt — OPTForQuestionAnswering(OPTConfig モデル)
- qwen2 — Qwen2ForQuestionAnswering(Qwen2Config モデル)
- qwen2_moe — Qwen2MoeForQuestionAnswering(Qwen2MoeConfig モデル)
- qwen3 — Qwen3ForQuestionAnswering(Qwen3Config モデル)
- qwen3_moe — Qwen3MoeForQuestionAnswering(Qwen3MoeConfig モデル)
- qwen3_next — Qwen3NextForQuestionAnswering(Qwen3NextConfig モデル)
- reformer — ReformerForQuestionAnswering(ReformerConfig モデル)
- rembert — RemBertForQuestionAnswering(RemBertConfig モデル)
- roberta — RobertaForQuestionAnswering(RobertaConfig モデル)
- roberta-prelayernorm — RobertaPreLayerNormForQuestionAnswering(RobertaPreLayerNormConfig モデル)
- roc_bert — RoCBertForQuestionAnswering(RoCBertConfig モデル)
- roformer — RoFormerForQuestionAnswering(RoFormerConfigモデル)
- seed_oss — SeedOssForQuestionAnswering(SeedOssConfig モデル)
- smollm3 — SmolLM3ForQuestionAnswering(SmolLM3Config モデル)
- splinter — SplinterForQuestionAnswering (SplinterConfig モデル)
- squeezebert — SqueezeBertForQuestionAnswering (SqueezeBertConfig モデル)
- t5 — T5ForQuestionAnswering(T5Config モデル)
- ultrabert — UltrabertForQuestionAnswering(UltrabertConfigモデル)
- umt5 — UMT5ForQuestionAnswering(UMT5Config モデル)
- xlm — XLMForQuestionAnsweringSimple (XLMConfig モデル)
- xlm-roberta — XLMRobertaForQuestionAnswering(XLMRobertaConfig モデル)
- xlm-roberta-xl — XLMRobertaXLForQuestionAnswering(XLMRobertaXLConfig モデル)
- xlnet — XLNetForQuestionAnsweringSimple(XLNetConfig モデル)
- xmod — XmodForQuestionAnswering(XmodConfigモデル)
- yoso — YosoForQuestionAnswering(YosoConfigモデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForQuestionAnswering
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForQuestionAnswering.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForQuestionAnswering.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForTextEncoding
コンピュータビジョン
以下のコンピュータビジョンタスクには、以下のautoクラスが利用可能です。
AutoModelForDepthEstimation
これは汎用的なモデルクラスであり、from_pretrained() クラスメソッドまたは from_config() クラスメソッドで生成された場合、ライブラリのモデルクラスの一つ(深度推定ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- CHMv2Config 設定クラス: CHMv2ForDepthEstimation (CHMv2Config モデル)
- DPTConfig 設定クラス: DPTForDepthEstimation (DPTConfig モデル)
- DepthAnythingConfig 設定クラス: DepthAnythingForDepthEstimation (DepthAnythingConfig モデル)
- DepthProConfig 設定クラス: DepthProForDepthEstimation (DepthProConfig モデル)
- GLPNConfig 設定クラス: GLPNForDepthEstimation (GLPNConfig モデル)
- PromptDepthAnythingConfig 設定クラス: PromptDepthAnythingForDepthEstimation (PromptDepthAnythingConfig モデル)
- Tipsv2DptConfig 設定クラス: Tipsv2DptForDepthEstimation (Tipsv2DptConfig モデル)
- ZoeDepthConfig 設定クラス: ZoeDepthForDepthEstimation (ZoeDepthConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(深度推定ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
ライブラリのモデルクラスの一つ(深度推定ヘッド付き)を、事前学習済みモデルからインスタンス化します。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- chmv2 — CHMv2ForDepthEstimation(CHMv2Config モデル)
- depth_anything — DepthAnythingForDepthEstimation(DepthAnythingConfig モデル)
- depth_pro — DepthProForDepthEstimation(DepthProConfig モデル)
- dpt — DPTForDepthEstimation(DPTConfig モデル)
- glpn — GLPNForDepthEstimation(GLPNConfig モデル)
- prompt_depth_anything — PromptDepthAnythingForDepthEstimation(PromptDepthAnythingConfig モデル)
- tipsv2_dpt — Tipsv2DptForDepthEstimation(Tipsv2DptConfig モデル)
- zoedepth — ZoeDepthForDepthEstimation(ZoeDepthConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForDepthEstimation
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForDepthEstimation.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForDepthEstimation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForNormalEstimation
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、通常の推定ヘッドを持つライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- Sapiens2Config 設定クラス: Sapiens2ForNormalEstimation (Sapiens2Config モデル)
- Tipsv2DptConfig 設定クラス: Tipsv2DptForNormalEstimation (Tipsv2DptConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(通常の推定ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- sapiens2 — Sapiens2ForNormalEstimation(Sapiens2Config モデル)
- tipsv2_dpt — Tipsv2DptForNormalEstimation(Tipsv2DptConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForNormalEstimation
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForNormalEstimation.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForNormalEstimation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForPointmapEstimation
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスの一つ(pointmap推定ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- Sapiens2Config 設定クラス: Sapiens2ForPointmapEstimation (Sapiens2Config モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(pointmap推定ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
ライブラリのモデルクラスの一つ(pointmap推定ヘッド付き)を、事前学習済みモデルからインスタンス化します。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- sapiens2 — Sapiens2ForPointmapEstimation(Sapiens2Config モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForPointmapEstimation
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForPointmapEstimation.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForPointmapEstimation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForImageMatting
This is a generic model class that will be instantiated as one of the model classes of the library (with a image matting head) when created with the from_pretrained() class method or the from_config() class method.
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- Sapiens2Config 設定クラス: Sapiens2ForImageMatting (Sapiens2Config モデル)
- VitMatteConfig 設定クラス: VitMatteForImageMatting (VitMatteConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(image matting head付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- sapiens2 — Sapiens2ForImageMatting(Sapiens2Config モデル)
- vitmatte — VitMatteForImageMatting(VitMatteConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForImageMatting
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForImageMatting.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForImageMatting.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForTextRecognition
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、テキスト認識ヘッドを備えたライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- PPOCRV5MobileRecConfig 設定クラス: PPOCRV5MobileRecForTextRecognition (PPOCRV5MobileRecConfig モデル)
- PPOCRV5ServerRecConfig 設定クラス: PPOCRV5ServerRecForTextRecognition (PPOCRV5ServerRecConfig モデル)
- PPOCRV6SmallRecConfig 設定クラス: PPOCRV6SmallRecForTextRecognition (PPOCRV6SmallRecConfig モデル)
- PPOCRV6TinyRecConfig 設定クラス: PPOCRV6TinyRecForTextRecognition (PPOCRV6TinyRecConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(テキスト認識ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- pp_ocrv5_mobile_rec — PPOCRV5MobileRecForTextRecognition(PPOCRV5MobileRecConfig モデル)
- pp_ocrv5_server_rec — PPOCRV5ServerRecForTextRecognition(PPOCRV5ServerRecConfig モデル)
- pp_ocrv6_small_rec — PPOCRV6SmallRecForTextRecognition(PPOCRV6SmallRecConfig モデル)
- pp_ocrv6_tiny_rec — PPOCRV6TinyRecForTextRecognition(PPOCRV6TinyRecConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForTextRecognition
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForTextRecognition.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForTextRecognition.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForTableRecognition
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、テーブル認識ヘッドを持つライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- SLANeXtConfig 設定クラス: SLANeXtForTableRecognition (SLANeXtConfig モデル)
- SLANetConfig 設定クラス: SLANetForTableRecognition (SLANetConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(テーブル認識ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- slanet — SLANetForTableRecognition(SLANetConfigモデル)
- slanext — SLANeXtForTableRecognition(SLANeXtConfigモデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForTableRecognition
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForTableRecognition.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForTableRecognition.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForImageClassification
これは汎用のモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスの一つ(画像分類ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- BeitConfig 設定クラス: BeitForImageClassification (BeitConfig モデル)
- BitConfig 設定クラス: BitForImageClassification (BitConfig モデル)
- CLIPConfig 設定クラス: CLIPForImageClassification (CLIPConfig モデル)
- ConvNextConfig 設定クラス: ConvNextForImageClassification (ConvNextConfig モデル)
- ConvNextV2Config 設定クラス: ConvNextV2ForImageClassification (ConvNextV2Config モデル)
- CvtConfig 設定クラス: CvtForImageClassification (CvtConfig モデル)
- Data2VecVisionConfig 設定クラス: Data2VecVisionForImageClassification (Data2VecVisionConfig モデル)
- DeiTConfig 設定クラス: DeiTForImageClassification または DeiTForImageClassificationWithTeacher (DeiTConfig モデル)
- DinatConfig 設定クラス: DinatForImageClassification (DinatConfig モデル)
- Dinov2Config 設定クラス: Dinov2ForImageClassification (Dinov2Config モデル)
- Dinov2WithRegistersConfig 設定クラス: Dinov2WithRegistersForImageClassification (Dinov2WithRegistersConfig モデル)
- DonutSwinConfig 設定クラス: DonutSwinForImageClassification (DonutSwinConfig モデル)
- EfficientNetConfig 設定クラス: EfficientNetForImageClassification (EfficientNetConfig モデル)
- FocalNetConfig 設定クラス: FocalNetForImageClassification (FocalNetConfig モデル)
- HGNetV2Config 設定クラス: HGNetV2ForImageClassification (HGNetV2Config モデル)
- HieraConfig 設定クラス: HieraForImageClassification (HieraConfig モデル)
- IJepaConfig 設定クラス: IJepaForImageClassification (IJepaConfig モデル)
- ImageGPTConfig 設定クラス: ImageGPTForImageClassification (ImageGPTConfig モデル)
- LevitConfig 設定クラス: LevitForImageClassification または LevitForImageClassificationWithTeacher (LevitConfig モデル)
- MetaClip2Config 設定クラス: MetaClip2ForImageClassification (MetaClip2Config モデル)
- MobileNetV1Config 設定クラス: MobileNetV1ForImageClassification (MobileNetV1Config モデル)
- MobileNetV2Config 設定クラス: MobileNetV2ForImageClassification (MobileNetV2Config モデル)
- MobileViTConfig 設定クラス: MobileViTForImageClassification (MobileViTConfig モデル)
- MobileViTV2Config 設定クラス: MobileViTV2ForImageClassification (MobileViTV2Config モデル)
- PPLCNetConfig 設定クラス: PPLCNetForImageClassification (PPLCNetConfig モデル)
- PerceiverConfig 設定クラス: PerceiverForImageClassificationLearned または PerceiverForImageClassificationFourier または PerceiverForImageClassificationConvProcessing (PerceiverConfig モデル)
- PoolFormerConfig 設定クラス: PoolFormerForImageClassification (PoolFormerConfig モデル)
- PvtConfig 設定クラス: PvtForImageClassification (PvtConfig モデル)
- PvtV2Config 設定クラス: PvtV2ForImageClassification (PvtV2Config モデル)
- RegNetConfig 設定クラス: RegNetForImageClassification (RegNetConfig モデル)
- ResNetConfig 設定クラス: ResNetForImageClassification (ResNetConfig モデル)
- SegformerConfig 設定クラス: SegformerForImageClassification (SegformerConfig モデル)
- ShieldGemma2Config 設定クラス: ShieldGemma2ForImageClassification (ShieldGemma2Config モデル)
- Siglip2Config 設定クラス: Siglip2ForImageClassification (Siglip2Config モデル)
- SiglipConfig 設定クラス: SiglipForImageClassification (SiglipConfig モデル)
- SwiftFormerConfig 設定クラス: SwiftFormerForImageClassification (SwiftFormerConfig モデル)
- SwinConfig 設定クラス: SwinForImageClassification (SwinConfig モデル)
- Swinv2Config 設定クラス: Swinv2ForImageClassification (Swinv2Config モデル)
- TextNetConfig 設定クラス: TextNetForImageClassification (TextNetConfig モデル)
- TimmWrapperConfig 設定クラス: TimmWrapperForImageClassification (TimmWrapperConfig モデル)
- ViTConfig 設定クラス: ViTForImageClassification (ViTConfig モデル)
- ViTMSNConfig 設定クラス: ViTMSNForImageClassification (ViTMSNConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(画像分類ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- beit — BeitForImageClassification(BeitConfigモデル)
- bit — BitForImageClassification (BitConfig モデル)
- clip — CLIPForImageClassification(CLIPConfig モデル)
- convnext — ConvNextForImageClassification(ConvNextConfig モデル)
- convnextv2 — ConvNextV2ForImageClassification(ConvNextV2Config モデル)
- cvt — CvtForImageClassification(CvtConfigモデル)
- data2vec-vision — Data2VecVisionForImageClassification(Data2VecVisionConfig モデル)
- deit — DeiTForImageClassification または DeiTForImageClassificationWithTeacher(DeiTConfig モデル)
- dinat — DinatForImageClassification(DinatConfigモデル)
- dinov2 — Dinov2ForImageClassification (Dinov2Config モデル)
- dinov2_with_registers — Dinov2WithRegistersForImageClassification(Dinov2WithRegistersConfig モデル)
- donut-swin — DonutSwinForImageClassification(DonutSwinConfig モデル)
- efficientnet — EfficientNetForImageClassification(EfficientNetConfig モデル)
- focalnet — FocalNetForImageClassification (FocalNetConfig モデル)
- hgnet_v2 — HGNetV2ForImageClassification(HGNetV2Config モデル)
- hiera — HieraForImageClassification (HieraConfig モデル)
- ijepa — IJepaForImageClassification(IJepaConfigモデル)
- imagegpt — ImageGPTForImageClassification(ImageGPTConfig モデル)
- levit — LevitForImageClassification または LevitForImageClassificationWithTeacher(LevitConfig モデル)
- metaclip_2 — MetaClip2ForImageClassification(MetaClip2Config モデル)
- mobilenet_v1 — MobileNetV1ForImageClassification(MobileNetV1Config モデル)
- mobilenet_v2 — MobileNetV2ForImageClassification(MobileNetV2Config モデル)
- mobilevit — MobileViTForImageClassification (MobileViTConfig モデル)
- mobilevitv2 — MobileViTV2ForImageClassification(MobileViTV2Config モデル)
- perceiver — PerceiverForImageClassificationLearned または PerceiverForImageClassificationFourier または PerceiverForImageClassificationConvProcessing (PerceiverConfig モデル)
- poolformer — PoolFormerForImageClassification(PoolFormerConfig モデル)
- pp_lcnet — PPLCNetForImageClassification(PPLCNetConfigモデル)
- pvt — PvtForImageClassification (PvtConfig モデル)
- pvt_v2 — PvtV2ForImageClassification (PvtV2Config モデル)
- regnet — RegNetForImageClassification(RegNetConfig モデル)
- resnet — ResNetForImageClassification(ResNetConfig モデル)
- segformer — SegformerForImageClassification (SegformerConfig モデル)
- shieldgemma2 — ShieldGemma2ForImageClassification(ShieldGemma2Config モデル)
- siglip — SiglipForImageClassification(SiglipConfig モデル)
- siglip2 — Siglip2ForImageClassification(Siglip2Config モデル)
- swiftformer — SwiftFormerForImageClassification(SwiftFormerConfigモデル)
- swin — SwinForImageClassification(SwinConfig モデル)
- swinv2 — Swinv2ForImageClassification(Swinv2Config モデル)
- textnet — TextNetForImageClassification (TextNetConfig モデル)
- timm_wrapper — TimmWrapperForImageClassification(TimmWrapperConfig モデル)
- vit — ViTForImageClassification (ViTConfig モデル)
- vit_msn — ViTMSNForImageClassification(ViTMSNConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForImageClassification
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForImageClassification.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForImageClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForVideoClassification
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスの一つ(動画分類ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- TimesformerConfig 設定クラス: TimesformerForVideoClassification (TimesformerConfig モデル)
- VJEPA2Config 設定クラス: VJEPA2ForVideoClassification (VJEPA2Config モデル)
- VideoMAEConfig 設定クラス: VideoMAEForVideoClassification (VideoMAEConfig モデル)
- VideoPrismVisionConfig 設定クラス: VideoPrismForVideoClassification (VideoPrismVisionConfig モデル)
- VivitConfig 設定クラス: VivitForVideoClassification (VivitConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(動画分類ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
ライブラリのモデルクラスの一つ(動画分類ヘッド付き)を事前学習済みモデルからインスタンス化します。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- timesformer — TimesformerForVideoClassification(TimesformerConfigモデル)
- videomae — VideoMAEForVideoClassification(VideoMAEConfig モデル)
- videoprism_vision_model — VideoPrismForVideoClassification(VideoPrismVisionConfig モデル)
- vivit — VivitForVideoClassification(VivitConfigモデル)
- vjepa2 — VJEPA2ForVideoClassification(VJEPA2Config モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForVideoClassification
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForVideoClassification.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForVideoClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForPoseEstimation
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスの一つ(ポーズ推定ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- Sapiens2Config 設定クラス: Sapiens2ForPoseEstimation (Sapiens2Config モデル)
- VitPoseConfig 設定クラス: VitPoseForPoseEstimation (VitPoseConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(ポーズ推定ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- sapiens2 — Sapiens2ForPoseEstimation(Sapiens2Config モデル)
- vitpose — VitPoseForPoseEstimation (VitPoseConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForPoseEstimation
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForPoseEstimation.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForPoseEstimation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForKeypointDetection
AutoModelForKeypointMatching
AutoModelForMaskedImageModeling
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、マスクドイメージモデリングヘッドを持つライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- DeiTConfig 設定クラス: DeiTForMaskedImageModeling (DeiTConfig モデル)
- FocalNetConfig 設定クラス: FocalNetForMaskedImageModeling (FocalNetConfig モデル)
- SwinConfig 設定クラス: SwinForMaskedImageModeling (SwinConfig モデル)
- Swinv2Config 設定クラス: Swinv2ForMaskedImageModeling (Swinv2Config モデル)
- ViTConfig 設定クラス: ViTForMaskedImageModeling (ViTConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(マスクドイメージモデリングヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- deit — DeiTForMaskedImageModeling(DeiTConfig モデル)
- focalnet — FocalNetForMaskedImageModeling(FocalNetConfigモデル)
- swin — SwinForMaskedImageModeling(SwinConfig モデル)
- swinv2 — Swinv2ForMaskedImageModeling (Swinv2Config モデル)
- vit — ViTForMaskedImageModeling (ViTConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForMaskedImageModeling
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForMaskedImageModeling.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForMaskedImageModeling.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForObjectDetection
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、オブジェクト検出ヘッドを持つライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- ConditionalDetrConfig 設定クラス: ConditionalDetrForObjectDetection (ConditionalDetrConfig モデル)
- DFineConfig 設定クラス: DFineForObjectDetection (DFineConfig モデル)
- DabDetrConfig 設定クラス: DabDetrForObjectDetection (DabDetrConfig モデル)
- DeformableDetrConfig 設定クラス: DeformableDetrForObjectDetection (DeformableDetrConfig モデル)
- Deimv2Config 設定クラス: Deimv2ForObjectDetection (Deimv2Config モデル)
- DetrConfig 設定クラス: DetrForObjectDetection (DetrConfig モデル)
- LwDetrConfig 設定クラス: LwDetrForObjectDetection (LwDetrConfig モデル)
- PPDocLayoutV2Config 設定クラス: PPDocLayoutV2ForObjectDetection (PPDocLayoutV2Config モデル)
- PPDocLayoutV3Config 設定クラス: PPDocLayoutV3ForObjectDetection (PPDocLayoutV3Config モデル)
- PPDocLayoutV4Config 設定クラス: PPDocLayoutV4ForObjectDetection (PPDocLayoutV4Config モデル)
- PPOCRV5MobileDetConfig 設定クラス: PPOCRV5MobileDetForObjectDetection (PPOCRV5MobileDetConfig モデル)
- PPOCRV5ServerDetConfig 設定クラス: PPOCRV5ServerDetForObjectDetection (PPOCRV5ServerDetConfig モデル)
- PPOCRV6MediumDetConfig 設定クラス: PPOCRV6MediumDetForObjectDetection (PPOCRV6MediumDetConfig モデル)
- PPOCRV6SmallDetConfig 設定クラス: PPOCRV6SmallDetForObjectDetection (PPOCRV6SmallDetConfig モデル)
- RTDetrConfig 設定クラス: RTDetrForObjectDetection (RTDetrConfig モデル)
- RTDetrV2Config 設定クラス: RTDetrV2ForObjectDetection (RTDetrV2Config モデル)
- RfDetrConfig 設定クラス: RfDetrForObjectDetection (RfDetrConfig モデル)
- TableTransformerConfig 設定クラス: TableTransformerForObjectDetection (TableTransformerConfig モデル)
- YolosConfig 設定クラス: YolosForObjectDetection (YolosConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(オブジェクト検出ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
ライブラリのモデルクラスの一つ(object detection head付き)を、事前学習済みモデルからインスタンス化します。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- conditional_detr — ConditionalDetrForObjectDetection(ConditionalDetrConfig モデル)
- d_fine — DFineForObjectDetection(DFineConfigモデル)
- dab-detr — DabDetrForObjectDetection(DabDetrConfig モデル)
- deformable_detr — DeformableDetrForObjectDetection(DeformableDetrConfig モデル)
- deimv2 — Deimv2ForObjectDetection(Deimv2Config モデル)
- detr — DetrForObjectDetection(DetrConfigモデル)
- lw_detr — LwDetrForObjectDetection(LwDetrConfig モデル)
- pp_doclayout_v2 — PPDocLayoutV2ForObjectDetection(PPDocLayoutV2Configモデル)
- pp_doclayout_v3 — PPDocLayoutV3ForObjectDetection(PPDocLayoutV3Config モデル)
- pp_doclayout_v4 — PPDocLayoutV4ForObjectDetection(PPDocLayoutV4Config モデル)
- pp_ocrv5_mobile_det — PPOCRV5MobileDetForObjectDetection(PPOCRV5MobileDetConfig モデル)
- pp_ocrv5_server_det — PPOCRV5ServerDetForObjectDetection(PPOCRV5ServerDetConfig モデル)
- pp_ocrv6_medium_det — PPOCRV6MediumDetForObjectDetection(PPOCRV6MediumDetConfig モデル)
- pp_ocrv6_small_det — PPOCRV6SmallDetForObjectDetection(PPOCRV6SmallDetConfig モデル)
- rf_detr — RfDetrForObjectDetection(RfDetrConfig モデル)
- rt_detr — RTDetrForObjectDetection(RTDetrConfigモデル)
- rt_detr_v2 — RTDetrV2ForObjectDetection(RTDetrV2Config モデル)
- table-transformer — TableTransformerForObjectDetection(TableTransformerConfig モデル)
- yolos — YolosForObjectDetection(YolosConfigモデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForObjectDetection
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForObjectDetection.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForObjectDetection.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForImageSegmentation
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスの一つ(画像セグメンテーションヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- DetrConfig 設定クラス: DetrForSegmentation (DetrConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(画像セグメンテーションヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- detr — DetrForSegmentation(DetrConfigモデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForImageSegmentation
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForImageSegmentation.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForImageSegmentation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForImageToImage
AutoModelForSemanticSegmentation
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成されると、セマンティックセグメンテーションヘッド付きのライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- BeitConfig 設定クラス: BeitForSemanticSegmentation (BeitConfig モデル)
- DPTConfig 設定クラス: DPTForSemanticSegmentation (DPTConfig モデル)
- Data2VecVisionConfig 設定クラス: Data2VecVisionForSemanticSegmentation (Data2VecVisionConfig モデル)
- MobileNetV2Config 設定クラス: MobileNetV2ForSemanticSegmentation (MobileNetV2Config モデル)
- MobileViTConfig 設定クラス: MobileViTForSemanticSegmentation (MobileViTConfig モデル)
- MobileViTV2Config 設定クラス: MobileViTV2ForSemanticSegmentation (MobileViTV2Config モデル)
- Sapiens2Config 設定クラス: Sapiens2ForSemanticSegmentation (Sapiens2Config モデル)
- SegformerConfig 設定クラス: SegformerForSemanticSegmentation (SegformerConfig モデル)
- Tipsv2DptConfig 設定クラス: Tipsv2DptForSemanticSegmentation (Tipsv2DptConfig モデル)
- UperNetConfig 設定クラス: UperNetForSemanticSegmentation (UperNetConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(セマンティックセグメンテーションヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
Transformers ライブラリのモデルクラスの一つ(セマンティックセグメンテーションヘッド付き)を、事前学習済みモデルからインスタンス化します。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- beit — BeitForSemanticSegmentation(BeitConfigモデル)
- data2vec-vision — Data2VecVisionForSemanticSegmentation(Data2VecVisionConfig モデル)
- dpt — DPTForSemanticSegmentation(DPTConfig モデル)
- mobilenet_v2 — MobileNetV2ForSemanticSegmentation(MobileNetV2Config モデル)
- mobilevit — MobileViTForSemanticSegmentation(MobileViTConfig モデル)
- mobilevitv2 — MobileViTV2ForSemanticSegmentation(MobileViTV2Config モデル)
- sapiens2 — Sapiens2ForSemanticSegmentation(Sapiens2Config モデル)
- segformer — SegformerForSemanticSegmentation(SegformerConfig モデル)
- tipsv2_dpt — Tipsv2DptForSemanticSegmentation(Tipsv2DptConfig モデル)
- upernet — UperNetForSemanticSegmentation(UperNetConfigモデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForSemanticSegmentation
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForSemanticSegmentation.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForSemanticSegmentation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForInstanceSegmentation
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、インスタンスセグメンテーションヘッド付きのライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- MaskFormerConfig 設定クラス: MaskFormerForInstanceSegmentation (MaskFormerConfig モデル)
- RfDetrConfig 設定クラス: RfDetrForInstanceSegmentation (RfDetrConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(インスタンスセグメンテーションヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- maskformer — MaskFormerForInstanceSegmentation(MaskFormerConfig モデル)
- rf_detr — RfDetrForInstanceSegmentation(RfDetrConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForInstanceSegmentation
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForInstanceSegmentation.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForInstanceSegmentation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForUniversalSegmentation
これは汎用のモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、このライブラリのモデルクラスのいずれか(汎用の画像セグメンテーションヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- DetrConfig 設定クラス: DetrForSegmentation (DetrConfig モデル)
- EomtConfig 設定クラス: EomtForUniversalSegmentation (EomtConfig モデル)
- EomtDinov3Config 設定クラス: EomtDinov3ForUniversalSegmentation (EomtDinov3Config モデル)
- Mask2FormerConfig 設定クラス: Mask2FormerForUniversalSegmentation (Mask2FormerConfig モデル)
- MaskFormerConfig 設定クラス: MaskFormerForInstanceSegmentation (MaskFormerConfig モデル)
- OneFormerConfig 設定クラス: OneFormerForUniversalSegmentation (OneFormerConfig モデル)
- VideomtConfig 設定クラス: VideomtForUniversalSegmentation (VideomtConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(汎用画像セグメンテーションヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- detr — DetrForSegmentation(DetrConfigモデル)
- eomt — EomtForUniversalSegmentation(EomtConfigモデル)
- eomt_dinov3 — EomtDinov3ForUniversalSegmentation(EomtDinov3Config モデル)
- mask2former — Mask2FormerForUniversalSegmentation(Mask2FormerConfig モデル)
- maskformer — MaskFormerForInstanceSegmentation(MaskFormerConfig モデル)
- oneformer — OneFormerForUniversalSegmentation(OneFormerConfig モデル)
- videomt — VideomtForUniversalSegmentation(VideomtConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForUniversalSegmentation
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForUniversalSegmentation.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForUniversalSegmentation.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForZeroShotImageClassification
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスの一つ(ゼロショット画像分類ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- AlignConfig 設定クラス: AlignModel (AlignConfig モデル)
- AltCLIPConfig 設定クラス: AltCLIPModel (AltCLIPConfig モデル)
- Blip2Config 設定クラス: Blip2ForImageTextRetrieval (Blip2Config モデル)
- BlipConfig 設定クラス: BlipModel (BlipConfig モデル)
- CLIPConfig 設定クラス: CLIPModel (CLIPConfig モデル)
- CLIPSegConfig 設定クラス: CLIPSegModel (CLIPSegConfig モデル)
- ChineseCLIPConfig 設定クラス: ChineseCLIPModel (ChineseCLIPConfig モデル)
- MetaClip2Config 設定クラス: MetaClip2Model (MetaClip2Config モデル)
- Siglip2Config 設定クラス: Siglip2Model (Siglip2Config モデル)
- SiglipConfig 設定クラス: SiglipModel (SiglipConfig モデル)
- Tipsv2Config 設定クラス: Tipsv2Model (Tipsv2Config モデル)
- VideoPrismConfig 設定クラス: VideoPrismClipModel (VideoPrismConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの1つ(ゼロショット画像分類ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- align — AlignModel(AlignConfig モデル)
- altclip — AltCLIPModel(AltCLIPConfig モデル)
- blip — BlipModel(BlipConfig モデル)
- blip-2 — Blip2ForImageTextRetrieval(Blip2Config モデル)
- chinese_clip — ChineseCLIPModel(ChineseCLIPConfig モデル)
- clip — CLIPModel (CLIPConfig モデル)
- clipseg — CLIPSegModel(CLIPSegConfig モデル)
- metaclip_2 — MetaClip2Model(MetaClip2Config モデル)
- siglip — SiglipModel(SiglipConfig モデル)
- siglip2 — Siglip2Model(Siglip2Config モデル)
- tipsv2 — Tipsv2Model(Tipsv2Config モデル)
- videoprism — VideoPrismClipModel(VideoPrismConfigモデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForZeroShotImageClassification
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForZeroShotImageClassification.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForZeroShotImageClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForZeroShotObjectDetection
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、このライブラリのモデルクラスのいずれか(ゼロショットオブジェクト検出ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- GroundingDinoConfig 設定クラス: GroundingDinoForObjectDetection (GroundingDinoConfig モデル)
- MMGroundingDinoConfig 設定クラス: MMGroundingDinoForObjectDetection (MMGroundingDinoConfig モデル)
- OmDetTurboConfig 設定クラス: OmDetTurboForObjectDetection (OmDetTurboConfig モデル)
- OwlViTConfig 設定クラス: OwlViTForObjectDetection (OwlViTConfig モデル)
- Owlv2Config 設定クラス: Owlv2ForObjectDetection (Owlv2Config モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(ゼロショットオブジェクト検出ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- grounding-dino — GroundingDinoForObjectDetection(GroundingDinoConfig モデル)
- mm-grounding-dino — MMGroundingDinoForObjectDetection(MMGroundingDinoConfigモデル)
- omdet-turbo — OmDetTurboForObjectDetection(OmDetTurboConfig モデル)
- owlv2 — Owlv2ForObjectDetection(Owlv2Config モデル)
- owlvit — OwlViTForObjectDetection(OwlViTConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForZeroShotObjectDetection
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForZeroShotObjectDetection.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForZeroShotObjectDetection.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
Trueオーディオ
以下の音声タスクには、以下のオートクラスが利用可能です。
AutoModelForAudioClassification
これは汎用的なモデルクラスであり、from_pretrained() クラスメソッドまたは from_config() クラスメソッドで生成された場合、このライブラリのモデルクラスのいずれか(音声分類ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- ASTConfig 設定クラス: ASTForAudioClassification (ASTConfig モデル)
- Data2VecAudioConfig 設定クラス: Data2VecAudioForSequenceClassification (Data2VecAudioConfig モデル)
- HubertConfig 設定クラス: HubertForSequenceClassification (HubertConfig モデル)
- SEWConfig 設定クラス: SEWForSequenceClassification (SEWConfig モデル)
- SEWDConfig 設定クラス: SEWDForSequenceClassification (SEWDConfig モデル)
- UniSpeechConfig 設定クラス: UniSpeechForSequenceClassification (UniSpeechConfig モデル)
- UniSpeechSatConfig 設定クラス: UniSpeechSatForSequenceClassification (UniSpeechSatConfig モデル)
- Wav2Vec2BertConfig 設定クラス: Wav2Vec2BertForSequenceClassification (Wav2Vec2BertConfig モデル)
- Wav2Vec2Config 設定クラス: Wav2Vec2ForSequenceClassification (Wav2Vec2Config モデル)
- Wav2Vec2ConformerConfig 設定クラス: Wav2Vec2ConformerForSequenceClassification (Wav2Vec2ConformerConfig モデル)
- WavLMConfig 設定クラス: WavLMForSequenceClassification (WavLMConfig モデル)
- WhisperConfig 設定クラス: WhisperForAudioClassification (WhisperConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(音声分類ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- audio-spectrogram-transformer — ASTForAudioClassification(ASTConfig モデル)
- data2vec-audio — Data2VecAudioForSequenceClassification(Data2VecAudioConfig モデル)
- hubert — HubertForSequenceClassification(HubertConfigモデル)
- sew — SEWForSequenceClassification(SEWConfigモデル)
- sew-d — SEWDForSequenceClassification(SEWDConfig モデル)
- unispeech — UniSpeechForSequenceClassification(UniSpeechConfigモデル)
- unispeech-sat — UniSpeechSatForSequenceClassification(UniSpeechSatConfig モデル)
- wav2vec2 — Wav2Vec2ForSequenceClassification(Wav2Vec2Config モデル)
- wav2vec2-bert — Wav2Vec2BertForSequenceClassification(Wav2Vec2BertConfig モデル)
- wav2vec2-conformer — Wav2Vec2ConformerForSequenceClassification(Wav2Vec2ConformerConfig モデル)
- wavlm — WavLMForSequenceClassification(WavLMConfig モデル)
- whisper — WhisperForAudioClassification(WhisperConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForAudioClassification
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForAudioClassification.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForAudioClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForAudioFrameClassification
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、このライブラリのモデルクラスの一つ(音声フレーム(トークン)分類ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- Data2VecAudioConfig 設定クラス: Data2VecAudioForAudioFrameClassification (Data2VecAudioConfig モデル)
- Nemotron3DiarizationConfig 設定クラス: Nemotron3DiarizationForAudioFrameClassification (Nemotron3DiarizationConfig モデル)
- UniSpeechSatConfig 設定クラス: UniSpeechSatForAudioFrameClassification (UniSpeechSatConfig モデル)
- Wav2Vec2BertConfig 設定クラス: Wav2Vec2BertForAudioFrameClassification (Wav2Vec2BertConfig モデル)
- Wav2Vec2Config 設定クラス: Wav2Vec2ForAudioFrameClassification (Wav2Vec2Config モデル)
- Wav2Vec2ConformerConfig 設定クラス: Wav2Vec2ConformerForAudioFrameClassification (Wav2Vec2ConformerConfig モデル)
- WavLMConfig 設定クラス: WavLMForAudioFrameClassification (WavLMConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの1つ(音声フレーム(トークン)分類ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
ライブラリのモデルクラスの一つを、事前学習済みモデルからインスタンス化します(音声フレーム(トークン)分類ヘッド付き)。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- data2vec-audio — Data2VecAudioForAudioFrameClassification(Data2VecAudioConfigモデル)
- nemotron3_diarization — Nemotron3DiarizationForAudioFrameClassification(Nemotron3DiarizationConfig モデル)
- unispeech-sat — UniSpeechSatForAudioFrameClassification(UniSpeechSatConfig モデル)
- wav2vec2 — Wav2Vec2ForAudioFrameClassification(Wav2Vec2Config モデル)
- wav2vec2-bert — Wav2Vec2BertForAudioFrameClassification(Wav2Vec2BertConfig モデル)
- wav2vec2-conformer — Wav2Vec2ConformerForAudioFrameClassification(Wav2Vec2ConformerConfig モデル)
- wavlm — WavLMForAudioFrameClassification(WavLMConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForAudioFrameClassification
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForAudioFrameClassification.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForAudioFrameClassification.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForCTC
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、接続時間分類ヘッドを持つライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- Data2VecAudioConfig 設定クラス: Data2VecAudioForCTC (Data2VecAudioConfig モデル)
- GraniteSpeech5CTCConfig 設定クラス: GraniteSpeech5ForCTC (GraniteSpeech5CTCConfig モデル)
- HubertConfig 設定クラス: HubertForCTC (HubertConfig モデル)
- LasrCTCConfig 設定クラス: LasrForCTC (LasrCTCConfig モデル)
- ParakeetCTCConfig 設定クラス: ParakeetForCTC (ParakeetCTCConfig モデル)
- SEWConfig 設定クラス: SEWForCTC (SEWConfig モデル)
- SEWDConfig 設定クラス: SEWDForCTC (SEWDConfig モデル)
- UniSpeechConfig 設定クラス: UniSpeechForCTC (UniSpeechConfig モデル)
- UniSpeechSatConfig 設定クラス: UniSpeechSatForCTC (UniSpeechSatConfig モデル)
- Wav2Vec2BertConfig 設定クラス: Wav2Vec2BertForCTC (Wav2Vec2BertConfig モデル)
- Wav2Vec2Config 設定クラス: Wav2Vec2ForCTC (Wav2Vec2Config モデル)
- Wav2Vec2ConformerConfig 設定クラス: Wav2Vec2ConformerForCTC (Wav2Vec2ConformerConfig モデル)
- WavLMConfig 設定クラス: WavLMForCTC (WavLMConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(接続時間分類ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- data2vec-audio — Data2VecAudioForCTC(Data2VecAudioConfig モデル)
- granite_speech5_ctc — GraniteSpeech5ForCTC(GraniteSpeech5CTCConfig モデル)
- hubert — HubertForCTC(HubertConfig モデル)
- lasr_ctc — LasrForCTC(LasrCTCConfig モデル)
- parakeet_ctc — ParakeetForCTC(ParakeetCTCConfig モデル)
- sew — SEWForCTC(SEWConfigモデル)
- sew-d — SEWDForCTC (SEWDConfig モデル)
- unispeech — UniSpeechForCTC(UniSpeechConfigモデル)
- unispeech-sat — UniSpeechSatForCTC(UniSpeechSatConfig モデル)
- wav2vec2 — Wav2Vec2ForCTC(Wav2Vec2Config モデル)
- wav2vec2-bert — Wav2Vec2BertForCTC(Wav2Vec2BertConfig モデル)
- wav2vec2-conformer — Wav2Vec2ConformerForCTC(Wav2Vec2ConformerConfig モデル)
- wavlm — WavLMForCTC (WavLMConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForCTC
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForCTC.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForCTC.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForTDT
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスのいずれか(トークンと持続時間のトランスダクサーヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- ParakeetTDTConfig 設定クラス: ParakeetForTDT (ParakeetTDTConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(トークンと持続時間のトランスダクサーヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- parakeet_tdt — ParakeetForTDT(ParakeetTDTConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForTDT
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForTDT.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForTDT.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForRNNT
This is a generic model class that will be instantiated as one of the model classes of the library (with a RNN transducer head) when created with the from_pretrained() class method or the from_config() class method.
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- Nemotron3_5AsrConfig 設定クラス: Nemotron3_5AsrForRNNT (Nemotron3_5AsrConfig モデル)
- NemotronAsrStreamingConfig 設定クラス: NemotronAsrStreamingForRNNT (NemotronAsrStreamingConfig モデル)
- ParakeetRNNTConfig 設定クラス: ParakeetForRNNT (ParakeetRNNTConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(RNN transducer head付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
ライブラリのモデルクラスの一つ(RNN transducer head付き)を、事前学習済みモデルからインスタンス化します。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- nemotron3_5_asr — Nemotron3_5AsrForRNNT(Nemotron3_5AsrConfig モデル)
- nemotron_asr_streaming — NemotronAsrStreamingForRNNT(NemotronAsrStreamingConfig モデル)
- parakeet_rnnt — ParakeetForRNNT(ParakeetRNNTConfigモデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForRNNT
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForRNNT.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForRNNT.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForSpeechSeq2Seq
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスのいずれか(シーケンスツーシーケンスの音声認識モデリングヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- CanaryConfig 設定クラス: CanaryForConditionalGeneration (CanaryConfig モデル)
- CohereAsrConfig 設定クラス: CohereAsrForConditionalGeneration (CohereAsrConfig モデル)
- DiaConfig 設定クラス: DiaForConditionalGeneration (DiaConfig モデル)
- FunAsrNanoConfig 設定クラス: FunAsrNanoForConditionalGeneration (FunAsrNanoConfig モデル)
- GraniteSpeechConfig 設定クラス: GraniteSpeechForConditionalGeneration (GraniteSpeechConfig モデル)
- GraniteSpeechPlusConfig 設定クラス: GraniteSpeechPlusForConditionalGeneration (GraniteSpeechPlusConfig モデル)
- KyutaiSpeechToTextConfig 設定クラス: KyutaiSpeechToTextForConditionalGeneration (KyutaiSpeechToTextConfig モデル)
- MoonshineConfig 設定クラス: MoonshineForConditionalGeneration (MoonshineConfig モデル)
- MoonshineStreamingConfig 設定クラス: MoonshineStreamingForConditionalGeneration (MoonshineStreamingConfig モデル)
- Pop2PianoConfig 設定クラス: Pop2PianoForConditionalGeneration (Pop2PianoConfig モデル)
- Qwen3ASRConfig 設定クラス: Qwen3ASRForConditionalGeneration (Qwen3ASRConfig モデル)
- SeamlessM4TConfig 設定クラス: SeamlessM4TForSpeechToText (SeamlessM4TConfig モデル)
- SeamlessM4Tv2Config 設定クラス: SeamlessM4Tv2ForSpeechToText (SeamlessM4Tv2Config モデル)
- Speech2TextConfig 設定クラス: Speech2TextForConditionalGeneration (Speech2TextConfig モデル)
- SpeechEncoderDecoderConfig 設定クラス: SpeechEncoderDecoderModel (SpeechEncoderDecoderConfig モデル)
- SpeechT5Config 設定クラス: SpeechT5ForSpeechToText (SpeechT5Config モデル)
- VibeVoiceAsrConfig 設定クラス: VibeVoiceAsrForConditionalGeneration (VibeVoiceAsrConfig モデル)
- VibeVoiceConfig 設定クラス: VibeVoiceForConditionalGeneration (VibeVoiceConfig モデル)
- VoxtralConfig 設定クラス: VoxtralForConditionalGeneration (VoxtralConfig モデル)
- VoxtralRealtimeConfig 設定クラス: VoxtralRealtimeForConditionalGeneration (VoxtralRealtimeConfig モデル)
- WhisperConfig 設定クラス: WhisperForConditionalGeneration (WhisperConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(シーケンスツーシーケンスの音声認識モデリングヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
ライブラリのモデルクラスの一つ(シーケンスツーシーケンスの音声認識モデリングヘッド付き)を、事前学習済みモデルからインスタンス化します。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- canary — CanaryForConditionalGeneration(CanaryConfigモデル)
- cohere_asr — CohereAsrForConditionalGeneration (CohereAsrConfig モデル)
- dia — DiaForConditionalGeneration(DiaConfig モデル)
- fun_asr_nano — FunAsrNanoForConditionalGeneration(FunAsrNanoConfig モデル)
- granite_speech — GraniteSpeechForConditionalGeneration(GraniteSpeechConfigモデル)
- granite_speech_plus — GraniteSpeechPlusForConditionalGeneration(GraniteSpeechPlusConfig モデル)
- kyutai_speech_to_text — KyutaiSpeechToTextForConditionalGeneration(KyutaiSpeechToTextConfig モデル)
- moonshine — MoonshineForConditionalGeneration(MoonshineConfig モデル)
- moonshine_streaming — MoonshineStreamingForConditionalGeneration(MoonshineStreamingConfig モデル)
- pop2piano — Pop2PianoForConditionalGeneration(Pop2PianoConfig モデル)
- qwen3_asr — Qwen3ASRForConditionalGeneration(Qwen3ASRConfig モデル)
- seamless_m4t — SeamlessM4TForSpeechToText(SeamlessM4TConfig モデル)
- seamless_m4t_v2 — SeamlessM4Tv2ForSpeechToText(SeamlessM4Tv2Config モデル)
- speech-encoder-decoder — SpeechEncoderDecoderModel(SpeechEncoderDecoderConfig モデル)
- speech_to_text — Speech2TextForConditionalGeneration(Speech2TextConfig モデル)
- speecht5 — SpeechT5ForSpeechToText (SpeechT5Config モデル)
- vibevoice — VibeVoiceForConditionalGeneration(VibeVoiceConfig モデル)
- vibevoice_asr — VibeVoiceAsrForConditionalGeneration(VibeVoiceAsrConfig モデル)
- voxtral — VoxtralForConditionalGeneration(VoxtralConfig モデル)
- voxtral_realtime — VoxtralRealtimeForConditionalGeneration(VoxtralRealtimeConfig モデル)
- whisper — WhisperForConditionalGeneration(WhisperConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForSpeechSeq2Seq
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForSpeechSeq2Seq.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForSpeechSeq2Seq.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForAudioXVector
This is a generic model class that will be instantiated as one of the model classes of the library (with a audio retrieval via x-vector head) when created with the from_pretrained() class method or the from_config() class method.
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- Data2VecAudioConfig 設定クラス: Data2VecAudioForXVector (Data2VecAudioConfig モデル)
- UniSpeechSatConfig 設定クラス: UniSpeechSatForXVector (UniSpeechSatConfig モデル)
- Wav2Vec2BertConfig 設定クラス: Wav2Vec2BertForXVector (Wav2Vec2BertConfig モデル)
- Wav2Vec2Config 設定クラス: Wav2Vec2ForXVector (Wav2Vec2Config モデル)
- Wav2Vec2ConformerConfig 設定クラス: Wav2Vec2ConformerForXVector (Wav2Vec2ConformerConfig モデル)
- WavLMConfig 設定クラス: WavLMForXVector (WavLMConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(x-vectorヘッドによる音声検索)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- data2vec-audio — Data2VecAudioForXVector(Data2VecAudioConfig モデル)
- unispeech-sat — UniSpeechSatForXVector(UniSpeechSatConfig モデル)
- wav2vec2 — Wav2Vec2ForXVector (Wav2Vec2Config モデル)
- wav2vec2-bert — Wav2Vec2BertForXVector(Wav2Vec2BertConfig モデル)
- wav2vec2-conformer — Wav2Vec2ConformerForXVector(Wav2Vec2ConformerConfig モデル)
- wavlm — WavLMForXVector(WavLMConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForAudioXVector
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForAudioXVector.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForAudioXVector.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForTextToSpectrogram
AutoModelForTextToWaveform
AutoModelForAudioTokenization
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ライブラリのモデルクラスのいずれかとして(codebooksヘッドによる音声トークナイズを備えて)インスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- DacConfig 設定クラス: DacModel (DacConfig モデル)
- HiggsAudioV2TokenizerConfig 設定クラス: HiggsAudioV2TokenizerModel (HiggsAudioV2TokenizerConfig モデル)
- NeuCodecConfig 設定クラス: NeuCodecModel (NeuCodecConfig モデル)
- PocketTTSAudioTokenizerConfig 設定クラス: PocketTTSAudioTokenizerModel (PocketTTSAudioTokenizerConfig モデル)
- VibeVoiceAcousticTokenizerConfig 設定クラス: VibeVoiceAcousticTokenizerModel (VibeVoiceAcousticTokenizerConfig モデル)
- Xcodec2Config 設定クラス: Xcodec2Model (Xcodec2Config モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの1つ(コードブックヘッドによる音声トークナイズを備えたもの)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- dac — DacModel(DacConfigモデル)
- higgs_audio_v2_tokenizer — HiggsAudioV2TokenizerModel(HiggsAudioV2TokenizerConfig モデル)
- neucodec — NeuCodecModel(NeuCodecConfig モデル)
- pocket_tts_audio_tokenizer — PocketTTSAudioTokenizerModel(PocketTTSAudioTokenizerConfig モデル)
- vibevoice_acoustic_tokenizer — VibeVoiceAcousticTokenizerModel(VibeVoiceAcousticTokenizerConfig モデル)
- xcodec2 — Xcodec2Model(Xcodec2Configモデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForAudioTokenization
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForAudioTokenization.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForAudioTokenization.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
Trueマルチモーダル
以下のマルチモーダルタスクには、以下のオートクラスが利用可能です。
AutoModelForMultimodalLM
これは汎用的なモデルクラスであり、from_pretrained() クラスメソッドまたは from_config() クラスメソッドで生成された場合、このライブラリのモデルクラスの一つ(マルチモーダル生成ヘッド付き)としてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- AriaConfig 設定クラス: AriaForConditionalGeneration (AriaConfig モデル)
- AyaVisionConfig 設定クラス: AyaVisionForConditionalGeneration (AyaVisionConfig モデル)
- Blip2Config 設定クラス: Blip2ForConditionalGeneration (Blip2Config モデル)
- BlipConfig 設定クラス: BlipForConditionalGeneration (BlipConfig モデル)
- ChameleonConfig 設定クラス: ChameleonForConditionalGeneration (ChameleonConfig モデル)
- Cohere2VisionConfig 設定クラス: Cohere2VisionForConditionalGeneration (Cohere2VisionConfig モデル)
- CohereCompassConfig 設定クラス: CohereCompassForConditionalGeneration (CohereCompassConfig モデル)
- Cosmos3EdgeConfig 設定クラス: Cosmos3EdgeForConditionalGeneration (Cosmos3EdgeConfig モデル)
- Cosmos3OmniConfig 設定クラス: Cosmos3OmniForConditionalGeneration (Cosmos3OmniConfig モデル)
- DeepseekOcr2Config 設定クラス: DeepseekOcr2ForConditionalGeneration (DeepseekOcr2Config モデル)
- DeepseekVLConfig 設定クラス: DeepseekVLForConditionalGeneration (DeepseekVLConfig モデル)
- DeepseekVLHybridConfig 設定クラス: DeepseekVLHybridForConditionalGeneration (DeepseekVLHybridConfig モデル)
- DiffusionGemmaConfig 設定クラス: DiffusionGemmaForBlockDiffusion (DiffusionGemmaConfig モデル)
- Dots3NoteConfig 設定クラス: Dots3NoteForConditionalGeneration (Dots3NoteConfig モデル)
- Emu3Config 設定クラス: Emu3ForConditionalGeneration (Emu3Config モデル)
- Ernie4_5_VLMoeConfig 設定クラス: Ernie4_5_VLMoeForConditionalGeneration (Ernie4_5_VLMoeConfig モデル)
- EvollaConfig 設定クラス: EvollaForProteinText2Text (EvollaConfig モデル)
- Exaone4_5_Config 設定クラス: Exaone4_5_ForConditionalGeneration (Exaone4_5_Config モデル)
- FastVlmConfig 設定クラス: FastVlmForConditionalGeneration (FastVlmConfig モデル)
- Florence2Config 設定クラス: Florence2ForConditionalGeneration (Florence2Config モデル)
- FunAsrNanoConfig 設定クラス: FunAsrNanoForConditionalGeneration (FunAsrNanoConfig モデル)
- FuyuConfig 設定クラス: FuyuForCausalLM (FuyuConfig モデル)
- Gemma3Config 設定クラス: Gemma3ForConditionalGeneration (Gemma3Config モデル)
- Gemma3nConfig 設定クラス: Gemma3nForConditionalGeneration (Gemma3nConfig モデル)
- Gemma4Config 設定クラス: Gemma4ForConditionalGeneration (Gemma4Config モデル)
- Gemma4UnifiedConfig 設定クラス: Gemma4UnifiedForConditionalGeneration (Gemma4UnifiedConfig モデル)
- GitConfig 設定クラス: GitForCausalLM (GitConfig モデル)
- Glm46VConfig 設定クラス: Glm46VForConditionalGeneration (Glm46VConfig モデル)
- Glm4vConfig 設定クラス: Glm4vForConditionalGeneration (Glm4vConfig モデル)
- Glm4vMoeConfig 設定クラス: Glm4vMoeForConditionalGeneration (Glm4vMoeConfig モデル)
- Glm5NextConfig 設定クラス: Glm5NextForConditionalGeneration (Glm5NextConfig モデル)
- GlmAsrConfig 設定クラス: GlmAsrForConditionalGeneration (GlmAsrConfig モデル)
- GlmOcrConfig 設定クラス: GlmOcrForConditionalGeneration (GlmOcrConfig モデル)
- GlmgaConfig 設定クラス: Glm46VForConditionalGeneration (GlmgaConfig モデル)
- GotOcr2Config 設定クラス: GotOcr2ForConditionalGeneration (GotOcr2Config モデル)
- Granite4VisionConfig 設定クラス: Granite4VisionForConditionalGeneration (Granite4VisionConfig モデル)
- GraniteDoclingConfig 設定クラス: GraniteDoclingForConditionalGeneration (GraniteDoclingConfig モデル)
- GraniteSpeechConfig 設定クラス: GraniteSpeechForConditionalGeneration (GraniteSpeechConfig モデル)
- GraniteSpeechPlusConfig 設定クラス: GraniteSpeechPlusForConditionalGeneration (GraniteSpeechPlusConfig モデル)
- HunYuanVLConfig 設定クラス: HunYuanVLForConditionalGeneration (HunYuanVLConfig モデル)
- HyperCLOVAXVisionV2Config 設定クラス: HyperCLOVAXVisionV2ForConditionalGeneration (HyperCLOVAXVisionV2Config モデル)
- Idefics2Config 設定クラス: Idefics2ForConditionalGeneration (Idefics2Config モデル)
- Idefics3Config 設定クラス: Idefics3ForConditionalGeneration (Idefics3Config モデル)
- IdeficsConfig 設定クラス: IdeficsForVisionText2Text (IdeficsConfig モデル)
- Inkling2Config 設定クラス: Inkling2ForConditionalGeneration (Inkling2Config モデル)
- InklingConfig 設定クラス: InklingForConditionalGeneration (InklingConfig モデル)
- InstructBlipConfig 設定クラス: InstructBlipForConditionalGeneration (InstructBlipConfig モデル)
- InstructBlipVideoConfig 設定クラス: InstructBlipVideoForConditionalGeneration (InstructBlipVideoConfig モデル)
- InternVLConfig 設定クラス: InternVLForConditionalGeneration (InternVLConfig モデル)
- JanusConfig 設定クラス: JanusForConditionalGeneration (JanusConfig モデル)
- Kimi_K25Config 設定クラス: Kimi_K25ForConditionalGeneration (Kimi_K25Config モデル)
- Kosmos2Config 設定クラス: Kosmos2ForConditionalGeneration (Kosmos2Config モデル)
- Kosmos2_5Config 設定クラス: Kosmos2_5ForConditionalGeneration (Kosmos2_5Config モデル)
- KyutaiSpeechToTextConfig 設定クラス: KyutaiSpeechToTextForConditionalGeneration (KyutaiSpeechToTextConfig モデル)
- Lfm2VlConfig 設定クラス: Lfm2VlForConditionalGeneration (Lfm2VlConfig モデル)
- Lfm3VlConfig 設定クラス: Lfm3VlForConditionalGeneration (Lfm3VlConfig モデル)
- LightOnOcrConfig 設定クラス: LightOnOcrForConditionalGeneration (LightOnOcrConfig モデル)
- Llama4Config 設定クラス: Llama4ForConditionalGeneration (Llama4Config モデル)
- LlavaConfig 設定クラス: LlavaForConditionalGeneration (LlavaConfig モデル)
- LlavaNextConfig 設定クラス: LlavaNextForConditionalGeneration (LlavaNextConfig モデル)
- LlavaNextVideoConfig 設定クラス: LlavaNextVideoForConditionalGeneration (LlavaNextVideoConfig モデル)
- LlavaOnevisionConfig 設定クラス: LlavaOnevisionForConditionalGeneration (LlavaOnevisionConfig モデル)
- MiniCPMOcrConfig 設定クラス: MiniCPMOcrForConditionalGeneration (MiniCPMOcrConfig モデル)
- MiniCPMV4_6Config 設定クラス: MiniCPMV4_6ForConditionalGeneration (MiniCPMV4_6Config モデル)
- MiniCPMV4_7Config 設定クラス: MiniCPMV4_7ForConditionalGeneration (MiniCPMV4_7Config モデル)
- MiniMaxM3VLConfig 設定クラス: MiniMaxM3SparseForConditionalGeneration (MiniMaxM3VLConfig モデル)
- Mistral3Config 設定クラス: Mistral3ForConditionalGeneration (Mistral3Config モデル)
- Mistral4Config 設定クラス: Mistral4ForCausalLM (Mistral4Config モデル)
- MistralLarge4Config 設定クラス: MistralLarge4ForConditionalGeneration (MistralLarge4Config モデル)
- MllamaConfig 設定クラス: MllamaForConditionalGeneration (MllamaConfig モデル)
- Molmo2Config 設定クラス: Molmo2ForConditionalGeneration (Molmo2Config モデル)
- MossTranscribeDiarizeConfig 設定クラス: MossTranscribeDiarizeForConditionalGeneration (MossTranscribeDiarizeConfig モデル)
- MuseGlimmerConfig 設定クラス: MuseGlimmerForConditionalGeneration (MuseGlimmerConfig モデル)
- MuseSparkConfig 設定クラス: MuseSparkForConditionalGeneration (MuseSparkConfig モデル)
- Nemotron3_5TranscribeConfig 設定クラス: Nemotron3_5TranscribeForConditionalGeneration (Nemotron3_5TranscribeConfig モデル)
- NemotronH_Omni_Reasoning_V3_Config 設定クラス: NemotronH_Omni_Reasoning_V3 (NemotronH_Omni_Reasoning_V3_Config モデル)
- Ovis2Config 設定クラス: Ovis2ForConditionalGeneration (Ovis2Config モデル)
- PI0Config 設定クラス: PI0ForConditionalGeneration (PI0Config モデル)
- PPChart2TableConfig 設定クラス: GotOcr2ForConditionalGeneration (PPChart2TableConfig モデル)
- PPFormulaNetConfig 設定クラス: PPFormulaNetForConditionalGeneration (PPFormulaNetConfig モデル)
- PaddleOCRVLConfig 設定クラス: PaddleOCRVLForConditionalGeneration (PaddleOCRVLConfig モデル)
- PaliGemmaConfig 設定クラス: PaliGemmaForConditionalGeneration (PaliGemmaConfig モデル)
- PerceptionLMConfig 設定クラス: PerceptionLMForConditionalGeneration (PerceptionLMConfig モデル)
- Phi4MultimodalConfig 設定クラス: Phi4MultimodalForCausalLM (Phi4MultimodalConfig モデル)
- Pix2StructConfig 設定クラス: Pix2StructForConditionalGeneration (Pix2StructConfig モデル)
- QianfanOCRConfig 設定クラス: QianfanOCRForConditionalGeneration (QianfanOCRConfig モデル)
- Qwen2AudioConfig 設定クラス: Qwen2AudioForConditionalGeneration (Qwen2AudioConfig モデル)
- Qwen2VLConfig 設定クラス: Qwen2VLForConditionalGeneration (Qwen2VLConfig モデル)
- Qwen2_5OmniConfig 設定クラス: Qwen2_5OmniForConditionalGeneration (Qwen2_5OmniConfig モデル)
- Qwen2_5OmniThinkerConfig 設定クラス: Qwen2_5OmniThinkerForConditionalGeneration (Qwen2_5OmniThinkerConfig モデル)
- Qwen2_5_VLConfig 設定クラス: Qwen2_5_VLForConditionalGeneration (Qwen2_5_VLConfig モデル)
- Qwen3ASRConfig 設定クラス: Qwen3ASRForConditionalGeneration (Qwen3ASRConfig モデル)
- Qwen3OmniMoeConfig 設定クラス: Qwen3OmniMoeForConditionalGeneration(Qwen3OmniMoeConfig モデル)
- Qwen3OmniMoeThinkerConfig 設定クラス: Qwen3OmniMoeThinkerForConditionalGeneration(Qwen3OmniMoeThinkerConfig モデル)
- Qwen3VLConfig 設定クラス: Qwen3VLForConditionalGeneration (Qwen3VLConfig モデル)
- Qwen3VLMoeConfig 設定クラス: Qwen3VLMoeForConditionalGeneration (Qwen3VLMoeConfig モデル)
- Qwen3_5Config 設定クラス: Qwen3_5ForConditionalGeneration (Qwen3_5Config モデル)
- Qwen3_5MoeConfig 設定クラス: Qwen3_5MoeForConditionalGeneration (Qwen3_5MoeConfig モデル)
- Qwen4ExpConfig 設定クラス: Qwen4ExpForConditionalGeneration (Qwen4ExpConfig モデル)
- ShieldGemma2Config 設定クラス: Gemma3ForConditionalGeneration (ShieldGemma2Config モデル)
- SmolVLMConfig 設定クラス: SmolVLMForConditionalGeneration (SmolVLMConfig モデル)
- Step3p7Config 設定クラス: Step3p7ForConditionalGeneration (Step3p7Config モデル)
- T5Gemma2Config 設定クラス: T5Gemma2ForConditionalGeneration (T5Gemma2Config モデル)
- UdopConfig 設定クラス: UdopForConditionalGeneration (UdopConfig モデル)
- UnlimitedOcrConfig 設定クラス: UnlimitedOcrForConditionalGeneration (UnlimitedOcrConfig モデル)
- VibeVoiceAsrConfig 設定クラス: VibeVoiceAsrForConditionalGeneration (VibeVoiceAsrConfig モデル)
- VideoLlama3Config 設定クラス: VideoLlama3ForConditionalGeneration (VideoLlama3Config モデル)
- VideoLlavaConfig 設定クラス: VideoLlavaForConditionalGeneration (VideoLlavaConfig モデル)
- VipLlavaConfig 設定クラス: VipLlavaForConditionalGeneration (VipLlavaConfig モデル)
- VisionEncoderDecoderConfig 設定クラス: VisionEncoderDecoderModel (VisionEncoderDecoderConfig モデル)
- VoxtralConfig 設定クラス: VoxtralForConditionalGeneration (VoxtralConfig モデル)
- VoxtralRealtimeConfig 設定クラス: VoxtralRealtimeForConditionalGeneration (VoxtralRealtimeConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(マルチモーダル生成ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- aria — AriaForConditionalGeneration(AriaConfig モデル)
- aya_vision — AyaVisionForConditionalGeneration(AyaVisionConfigモデル)
- blip — BlipForConditionalGeneration(BlipConfig モデル)
- blip-2 — Blip2ForConditionalGeneration(Blip2Config モデル)
- chameleon — ChameleonForConditionalGeneration(ChameleonConfig モデル)
- cohere2_vision — Cohere2VisionForConditionalGeneration(Cohere2VisionConfig モデル)
- cohere_compass — CohereCompassForConditionalGeneration(CohereCompassConfig モデル)
- cosmos3_edge — Cosmos3EdgeForConditionalGeneration(Cosmos3EdgeConfig モデル)
- cosmos3_omni — Cosmos3OmniForConditionalGeneration(Cosmos3OmniConfig モデル)
- deepseek_ocr2 — DeepseekOcr2ForConditionalGeneration (DeepseekOcr2Config モデル)
- deepseek_vl — DeepseekVLForConditionalGeneration(DeepseekVLConfig モデル)
- deepseek_vl_hybrid — DeepseekVLHybridForConditionalGeneration(DeepseekVLHybridConfig モデル)
- diffusion_gemma — DiffusionGemmaForBlockDiffusion(DiffusionGemmaConfig モデル)
- dots3_note — Dots3NoteForConditionalGeneration(Dots3NoteConfig モデル)
- emu3 — Emu3ForConditionalGeneration(Emu3Configモデル)
- ernie4_5_vl_moe — Ernie4_5_VLMoeForConditionalGeneration(Ernie4_5_VLMoeConfig モデル)
- evolla — EvollaForProteinText2Text(EvollaConfigモデル)
- exaone4_5 — Exaone4_5_ForConditionalGeneration(Exaone4_5_Config モデル)
- fast_vlm — FastVlmForConditionalGeneration(FastVlmConfigモデル)
- florence2 — Florence2ForConditionalGeneration(Florence2Config モデル)
- fun_asr_nano — FunAsrNanoForConditionalGeneration(FunAsrNanoConfig モデル)
- fuyu — FuyuForCausalLM(FuyuConfigモデル)
- gemma3 — Gemma3ForConditionalGeneration(Gemma3Config モデル)
- gemma3n — Gemma3nForConditionalGeneration(Gemma3nConfig モデル)
- gemma4 — Gemma4ForConditionalGeneration(Gemma4Config モデル)
- gemma4_unified — Gemma4UnifiedForConditionalGeneration(Gemma4UnifiedConfig モデル)
- git — GitForCausalLM(GitConfigモデル)
- glm46v — Glm46VForConditionalGeneration(Glm46VConfig モデル)
- glm4v — Glm4vForConditionalGeneration(Glm4vConfig モデル)
- glm4v_moe — Glm4vMoeForConditionalGeneration(Glm4vMoeConfig モデル)
- glm5_next — Glm5NextForConditionalGeneration(Glm5NextConfig モデル)
- glm_ocr — GlmOcrForConditionalGeneration(GlmOcrConfig モデル)
- glmasr — GlmAsrForConditionalGeneration(GlmAsrConfig モデル)
- glmga — Glm46VForConditionalGeneration(GlmgaConfig モデル)
- got_ocr2 — GotOcr2ForConditionalGeneration(GotOcr2Config モデル)
- granite4_vision — Granite4VisionForConditionalGeneration(Granite4VisionConfig モデル)
- granite_docling — GraniteDoclingForConditionalGeneration(GraniteDoclingConfigモデル)
- granite_speech — GraniteSpeechForConditionalGeneration(GraniteSpeechConfigモデル)
- granite_speech_plus — GraniteSpeechPlusForConditionalGeneration(GraniteSpeechPlusConfig モデル)
- hunyuan_vl — HunYuanVLForConditionalGeneration (HunYuanVLConfig モデル)
- hyperclovax_vision_v2 — HyperCLOVAXVisionV2ForConditionalGeneration(HyperCLOVAXVisionV2Config モデル)
- idefics — IdeficsForVisionText2Text(IdeficsConfigモデル)
- idefics2 — Idefics2ForConditionalGeneration(Idefics2Config モデル)
- idefics3 — Idefics3ForConditionalGeneration(Idefics3Config モデル)
- inkling2 — Inkling2ForConditionalGeneration(Inkling2Configモデル)
- inkling_mm_model — InklingForConditionalGeneration(InklingConfig モデル)
- instructblip — InstructBlipForConditionalGeneration(InstructBlipConfig モデル)
- instructblipvideo — InstructBlipVideoForConditionalGeneration(InstructBlipVideoConfig モデル)
- internvl — InternVLForConditionalGeneration(InternVLConfig モデル)
- janus — JanusForConditionalGeneration(JanusConfig モデル)
- kimi_k25 — Kimi_K25ForConditionalGeneration(Kimi_K25Config モデル)
- kosmos-2 — Kosmos2ForConditionalGeneration(Kosmos2Config モデル)
- kosmos-2.5 — Kosmos2_5ForConditionalGeneration(Kosmos2_5Config モデル)
- kyutai_speech_to_text — KyutaiSpeechToTextForConditionalGeneration(KyutaiSpeechToTextConfig モデル)
- lfm2_vl — Lfm2VlForConditionalGeneration(Lfm2VlConfig モデル)
- lfm3_vl — Lfm3VlForConditionalGeneration(Lfm3VlConfigモデル)
- lighton_ocr — LightOnOcrForConditionalGeneration(LightOnOcrConfigモデル)
- llama4 — Llama4ForConditionalGeneration(Llama4Config モデル)
- llava — LlavaForConditionalGeneration(LlavaConfigモデル)
- llava_next — LlavaNextForConditionalGeneration(LlavaNextConfig モデル)
- llava_next_video — LlavaNextVideoForConditionalGeneration(LlavaNextVideoConfig モデル)
- llava_onevision — LlavaOnevisionForConditionalGeneration(LlavaOnevisionConfig モデル)
- minicpm_ocr — MiniCPMOcrForConditionalGeneration(MiniCPMOcrConfig モデル)
- minicpmv4_6 — MiniCPMV4_6ForConditionalGeneration(MiniCPMV4_6Config モデル)
- minicpmv4_7 — MiniCPMV4_7ForConditionalGeneration(MiniCPMV4_7Config モデル)
- minimax_m3_vl — MiniMaxM3SparseForConditionalGeneration(MiniMaxM3VLConfig モデル)
- mistral3 — Mistral3ForConditionalGeneration(Mistral3Config モデル)
- mistral4 — Mistral4ForCausalLM(Mistral4Config モデル)
- mistral_large4 — MistralLarge4ForConditionalGeneration(MistralLarge4Config モデル)
- mllama — MllamaForConditionalGeneration(MllamaConfig モデル)
- molmo2 — Molmo2ForConditionalGeneration (Molmo2Config モデル)
- moss_transcribe_diarize — MossTranscribeDiarizeForConditionalGeneration(MossTranscribeDiarizeConfig モデル)
- muse_glimmer — MuseGlimmerForConditionalGeneration (MuseGlimmerConfig モデル)
- muse_spark — MuseSparkForConditionalGeneration(MuseSparkConfigモデル)
- nemo_speechlm — Nemotron3_5TranscribeForConditionalGeneration(Nemotron3_5TranscribeConfig モデル)
- nemotron_h_omni — NemotronH_Omni_Reasoning_V3(NemotronH_Omni_Reasoning_V3_Config モデル)
- ovis2 — Ovis2ForConditionalGeneration(Ovis2Config モデル)
- paddleocr_vl — PaddleOCRVLForConditionalGeneration(PaddleOCRVLConfig モデル)
- paligemma — PaliGemmaForConditionalGeneration(PaliGemmaConfig モデル)
- perception_lm — PerceptionLMForConditionalGeneration(PerceptionLMConfig モデル)
- phi4_multimodal — Phi4MultimodalForCausalLM(Phi4MultimodalConfig モデル)
- pi0 — PI0ForConditionalGeneration(PI0Configモデル)
- pix2struct — Pix2StructForConditionalGeneration(Pix2StructConfig モデル)
- pp_chart2table — GotOcr2ForConditionalGeneration(PPChart2TableConfig モデル)
- pp_formulanet — PPFormulaNetForConditionalGeneration(PPFormulaNetConfigモデル)
- qianfan_ocr — QianfanOCRForConditionalGeneration(QianfanOCRConfig モデル)
- qwen2_5_omni — Qwen2_5OmniForConditionalGeneration(Qwen2_5OmniConfig モデル)
- qwen2_5_omni_thinker — Qwen2_5OmniThinkerForConditionalGeneration (Qwen2_5OmniThinkerConfig モデル)
- qwen2_5_vl — Qwen2_5_VLForConditionalGeneration(Qwen2_5_VLConfig モデル)
- qwen2_audio — Qwen2AudioForConditionalGeneration(Qwen2AudioConfig モデル)
- qwen2_vl — Qwen2VLForConditionalGeneration(Qwen2VLConfigモデル)
- qwen3_5 — Qwen3_5ForConditionalGeneration(Qwen3_5Config モデル)
- qwen3_5_moe — Qwen3_5MoeForConditionalGeneration(Qwen3_5MoeConfig モデル)
- qwen3_asr — Qwen3ASRForConditionalGeneration(Qwen3ASRConfig モデル)
- qwen3_omni_moe — Qwen3OmniMoeForConditionalGeneration(Qwen3OmniMoeConfig モデル)
- qwen3_omni_moe_thinker — Qwen3OmniMoeThinkerForConditionalGeneration(Qwen3OmniMoeThinkerConfig モデル)
- qwen3_vl — Qwen3VLForConditionalGeneration(Qwen3VLConfig モデル)
- qwen3_vl_moe — Qwen3VLMoeForConditionalGeneration(Qwen3VLMoeConfig モデル)
- qwen4_exp — Qwen4ExpForConditionalGeneration(Qwen4ExpConfig モデル)
- shieldgemma2 — Gemma3ForConditionalGeneration(ShieldGemma2Config モデル)
- smolvlm — SmolVLMForConditionalGeneration(SmolVLMConfig モデル)
- step3p7 — Step3p7ForConditionalGeneration(Step3p7Config モデル)
- t5gemma2 — T5Gemma2ForConditionalGeneration(T5Gemma2Config モデル)
- udop — UdopForConditionalGeneration (UdopConfig モデル)
- unlimited_ocr — UnlimitedOcrForConditionalGeneration(UnlimitedOcrConfigモデル)
- vibevoice_asr — VibeVoiceAsrForConditionalGeneration(VibeVoiceAsrConfig モデル)
- video_llama_3 — VideoLlama3ForConditionalGeneration (VideoLlama3Config モデル)
- video_llava — VideoLlavaForConditionalGeneration(VideoLlavaConfigモデル)
- vipllava — VipLlavaForConditionalGeneration(VipLlavaConfig モデル)
- vision-encoder-decoder — VisionEncoderDecoderModel(VisionEncoderDecoderConfig モデル)
- voxtral — VoxtralForConditionalGeneration(VoxtralConfig モデル)
- voxtral_realtime — VoxtralRealtimeForConditionalGeneration(VoxtralRealtimeConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForMultimodalLM
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForMultimodalLM.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForMultimodalLM.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForTableQuestionAnswering
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、テーブル質問応答ヘッドを持つライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- TapasConfig 設定クラス: TapasForQuestionAnswering (TapasConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(テーブル質問応答ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
ライブラリのモデルクラスの一つ(テーブル質問応答ヘッド付き)を、事前学習済みモデルからインスタンス化します。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- tapas — TapasForQuestionAnswering(TapasConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForTableQuestionAnswering
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForTableQuestionAnswering.from_pretrained("google/tapas-base-finetuned-wtq")
>>> # Update configuration during loading
>>> model = AutoModelForTableQuestionAnswering.from_pretrained("google/tapas-base-finetuned-wtq", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForDocumentQuestionAnswering
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ドキュメント質問応答ヘッドを備えたライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- LayoutLMConfig 設定クラス: LayoutLMForQuestionAnswering (LayoutLMConfig モデル)
- LayoutLMv2Config 設定クラス: LayoutLMv2ForQuestionAnswering (LayoutLMv2Config モデル)
- LayoutLMv3Config 設定クラス: LayoutLMv3ForQuestionAnswering (LayoutLMv3Config モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(文書質問応答ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
例:
>>> from transformers import AutoConfig, AutoModelForDocumentQuestionAnswering
>>> # Download configuration from huggingface.co and cache.
>>> config = AutoConfig.from_pretrained("impira/layoutlm-document-qa", revision="52e01b3")
>>> model = AutoModelForDocumentQuestionAnswering.from_config(config)from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- layoutlm — LayoutLMForQuestionAnswering(LayoutLMConfigモデル)
- layoutlmv2 — LayoutLMv2ForQuestionAnswering(LayoutLMv2Config モデル)
- layoutlmv3 — LayoutLMv3ForQuestionAnswering(LayoutLMv3Config モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForDocumentQuestionAnswering
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForDocumentQuestionAnswering.from_pretrained("impira/layoutlm-document-qa", revision="52e01b3")
>>> # Update configuration during loading
>>> model = AutoModelForDocumentQuestionAnswering.from_pretrained("impira/layoutlm-document-qa", revision="52e01b3", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForVisualQuestionAnswering
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、ビジュアル質問応答ヘッドを備えたライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- Blip2Config 設定クラス: Blip2ForConditionalGeneration (Blip2Config モデル)
- BlipConfig 設定クラス: BlipForQuestionAnswering (BlipConfig モデル)
- ViltConfig 設定クラス: ViltForQuestionAnswering (ViltConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(ビジュアル質問応答ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- blip — BlipForQuestionAnswering(BlipConfig モデル)
- blip-2 — Blip2ForConditionalGeneration(Blip2Config モデル)
- vilt — ViltForQuestionAnswering(ViltConfigモデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForVisualQuestionAnswering
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForVisualQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
>>> # Update configuration during loading
>>> model = AutoModelForVisualQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForImageTextToText
This is a generic model class that will be instantiated as one of the model classes of the library (with a image-text-to-text modeling head) when created with the from_pretrained() class method or the from_config() class method.
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- AriaConfig 設定クラス: AriaForConditionalGeneration (AriaConfig モデル)
- AyaVisionConfig 設定クラス: AyaVisionForConditionalGeneration (AyaVisionConfig モデル)
- Blip2Config 設定クラス: Blip2ForConditionalGeneration (Blip2Config モデル)
- BlipConfig 設定クラス: BlipForConditionalGeneration (BlipConfig モデル)
- ChameleonConfig 設定クラス: ChameleonForConditionalGeneration (ChameleonConfig モデル)
- Cohere2VisionConfig 設定クラス: Cohere2VisionForConditionalGeneration (Cohere2VisionConfig モデル)
- CohereCompassConfig 設定クラス: CohereCompassForConditionalGeneration (CohereCompassConfig モデル)
- Cosmos3EdgeConfig 設定クラス: Cosmos3EdgeForConditionalGeneration (Cosmos3EdgeConfig モデル)
- Cosmos3OmniConfig 設定クラス: Cosmos3OmniForConditionalGeneration (Cosmos3OmniConfig モデル)
- DeepseekOcr2Config 設定クラス: DeepseekOcr2ForConditionalGeneration (DeepseekOcr2Config モデル)
- DeepseekVLConfig 設定クラス: DeepseekVLForConditionalGeneration (DeepseekVLConfig モデル)
- DeepseekVLHybridConfig 設定クラス: DeepseekVLHybridForConditionalGeneration (DeepseekVLHybridConfig モデル)
- DiffusionGemmaConfig 設定クラス: DiffusionGemmaForBlockDiffusion (DiffusionGemmaConfig モデル)
- Dots3NoteConfig 設定クラス: Dots3NoteForConditionalGeneration (Dots3NoteConfig モデル)
- Emu3Config 設定クラス: Emu3ForConditionalGeneration (Emu3Config モデル)
- Ernie4_5_VLMoeConfig 設定クラス: Ernie4_5_VLMoeForConditionalGeneration (Ernie4_5_VLMoeConfig モデル)
- EvollaConfig 設定クラス: EvollaForProteinText2Text (EvollaConfig モデル)
- Exaone4_5_Config 設定クラス: Exaone4_5_ForConditionalGeneration (Exaone4_5_Config モデル)
- FastVlmConfig 設定クラス: FastVlmForConditionalGeneration (FastVlmConfig モデル)
- Florence2Config 設定クラス: Florence2ForConditionalGeneration (Florence2Config モデル)
- FuyuConfig 設定クラス: FuyuForCausalLM (FuyuConfig モデル)
- Gemma3Config 設定クラス: Gemma3ForConditionalGeneration (Gemma3Config モデル)
- Gemma3nConfig 設定クラス: Gemma3nForConditionalGeneration (Gemma3nConfig モデル)
- Gemma4Config 設定クラス: Gemma4ForConditionalGeneration (Gemma4Config モデル)
- Gemma4UnifiedConfig 設定クラス: Gemma4UnifiedForConditionalGeneration (Gemma4UnifiedConfig モデル)
- GitConfig 設定クラス: GitForCausalLM (GitConfig モデル)
- Glm46VConfig 設定クラス: Glm46VForConditionalGeneration (Glm46VConfig モデル)
- Glm4vConfig 設定クラス: Glm4vForConditionalGeneration (Glm4vConfig モデル)
- Glm4vMoeConfig 設定クラス: Glm4vMoeForConditionalGeneration (Glm4vMoeConfig モデル)
- Glm5NextConfig 設定クラス: Glm5NextForConditionalGeneration (Glm5NextConfig モデル)
- GlmOcrConfig 設定クラス: GlmOcrForConditionalGeneration (GlmOcrConfig モデル)
- GlmgaConfig 設定クラス: Glm46VForConditionalGeneration (GlmgaConfig モデル)
- GotOcr2Config 設定クラス: GotOcr2ForConditionalGeneration (GotOcr2Config モデル)
- Granite4VisionConfig 設定クラス: Granite4VisionForConditionalGeneration (Granite4VisionConfig モデル)
- GraniteDoclingConfig 設定クラス: GraniteDoclingForConditionalGeneration (GraniteDoclingConfig モデル)
- HunYuanVLConfig 設定クラス: HunYuanVLForConditionalGeneration (HunYuanVLConfig モデル)
- HyperCLOVAXVisionV2Config 設定クラス: HyperCLOVAXVisionV2ForConditionalGeneration (HyperCLOVAXVisionV2Config モデル)
- Idefics2Config 設定クラス: Idefics2ForConditionalGeneration (Idefics2Config モデル)
- Idefics3Config 設定クラス: Idefics3ForConditionalGeneration (Idefics3Config モデル)
- IdeficsConfig 設定クラス: IdeficsForVisionText2Text (IdeficsConfig モデル)
- Inkling2Config 設定クラス: Inkling2ForConditionalGeneration (Inkling2Config モデル)
- InklingConfig 設定クラス: InklingForConditionalGeneration (InklingConfig モデル)
- InstructBlipConfig 設定クラス: InstructBlipForConditionalGeneration (InstructBlipConfig モデル)
- InstructBlipVideoConfig 設定クラス: InstructBlipVideoForConditionalGeneration (InstructBlipVideoConfig モデル)
- InternVLConfig 設定クラス: InternVLForConditionalGeneration (InternVLConfig モデル)
- JanusConfig 設定クラス: JanusForConditionalGeneration (JanusConfig モデル)
- Kimi_K25Config 設定クラス: Kimi_K25ForConditionalGeneration (Kimi_K25Config モデル)
- Kosmos2Config 設定クラス: Kosmos2ForConditionalGeneration (Kosmos2Config モデル)
- Kosmos2_5Config 設定クラス: Kosmos2_5ForConditionalGeneration (Kosmos2_5Config モデル)
- Lfm2VlConfig 設定クラス: Lfm2VlForConditionalGeneration (Lfm2VlConfig モデル)
- Lfm3VlConfig 設定クラス: Lfm3VlForConditionalGeneration (Lfm3VlConfig モデル)
- LightOnOcrConfig 設定クラス: LightOnOcrForConditionalGeneration (LightOnOcrConfig モデル)
- Llama4Config 設定クラス: Llama4ForConditionalGeneration (Llama4Config モデル)
- LlavaConfig 設定クラス: LlavaForConditionalGeneration (LlavaConfig モデル)
- LlavaNextConfig 設定クラス: LlavaNextForConditionalGeneration (LlavaNextConfig モデル)
- LlavaNextVideoConfig 設定クラス: LlavaNextVideoForConditionalGeneration (LlavaNextVideoConfig モデル)
- LlavaOnevisionConfig 設定クラス: LlavaOnevisionForConditionalGeneration (LlavaOnevisionConfig モデル)
- MiniCPMOcrConfig 設定クラス: MiniCPMOcrForConditionalGeneration (MiniCPMOcrConfig モデル)
- MiniCPMV4_6Config 設定クラス: MiniCPMV4_6ForConditionalGeneration (MiniCPMV4_6Config モデル)
- MiniCPMV4_7Config 設定クラス: MiniCPMV4_7ForConditionalGeneration (MiniCPMV4_7Config モデル)
- MiniMaxM3VLConfig 設定クラス: MiniMaxM3SparseForConditionalGeneration (MiniMaxM3VLConfig モデル)
- Mistral3Config 設定クラス: Mistral3ForConditionalGeneration (Mistral3Config モデル)
- Mistral4Config 設定クラス: Mistral4ForCausalLM (Mistral4Config モデル)
- MistralLarge4Config 設定クラス: MistralLarge4ForConditionalGeneration (MistralLarge4Config モデル)
- MllamaConfig 設定クラス: MllamaForConditionalGeneration (MllamaConfig モデル)
- Molmo2Config 設定クラス: Molmo2ForConditionalGeneration (Molmo2Config モデル)
- MuseGlimmerConfig 設定クラス: MuseGlimmerForConditionalGeneration (MuseGlimmerConfig モデル)
- MuseSparkConfig 設定クラス: MuseSparkForConditionalGeneration (MuseSparkConfig モデル)
- NemotronH_Omni_Reasoning_V3_Config 設定クラス: NemotronH_Omni_Reasoning_V3 (NemotronH_Omni_Reasoning_V3_Config モデル)
- Ovis2Config 設定クラス: Ovis2ForConditionalGeneration (Ovis2Config モデル)
- PI0Config 設定クラス: PI0ForConditionalGeneration (PI0Config モデル)
- PPChart2TableConfig 設定クラス: GotOcr2ForConditionalGeneration (PPChart2TableConfig モデル)
- PPFormulaNetConfig 設定クラス: PPFormulaNetForConditionalGeneration (PPFormulaNetConfig モデル)
- PaddleOCRVLConfig 設定クラス: PaddleOCRVLForConditionalGeneration (PaddleOCRVLConfig モデル)
- PaliGemmaConfig 設定クラス: PaliGemmaForConditionalGeneration (PaliGemmaConfig モデル)
- PerceptionLMConfig 設定クラス: PerceptionLMForConditionalGeneration (PerceptionLMConfig モデル)
- Pix2StructConfig 設定クラス: Pix2StructForConditionalGeneration (Pix2StructConfig モデル)
- QianfanOCRConfig 設定クラス: QianfanOCRForConditionalGeneration (QianfanOCRConfig モデル)
- Qwen2VLConfig 設定クラス: Qwen2VLForConditionalGeneration (Qwen2VLConfig モデル)
- Qwen2_5OmniThinkerConfig 設定クラス: Qwen2_5OmniThinkerForConditionalGeneration (Qwen2_5OmniThinkerConfig モデル)
- Qwen2_5_VLConfig 設定クラス: Qwen2_5_VLForConditionalGeneration (Qwen2_5_VLConfig モデル)
- Qwen3OmniMoeThinkerConfig 設定クラス: Qwen3OmniMoeThinkerForConditionalGeneration(Qwen3OmniMoeThinkerConfig モデル)
- Qwen3VLConfig 設定クラス: Qwen3VLForConditionalGeneration (Qwen3VLConfig モデル)
- Qwen3VLMoeConfig 設定クラス: Qwen3VLMoeForConditionalGeneration (Qwen3VLMoeConfig モデル)
- Qwen3_5Config 設定クラス: Qwen3_5ForConditionalGeneration (Qwen3_5Config モデル)
- Qwen3_5MoeConfig 設定クラス: Qwen3_5MoeForConditionalGeneration (Qwen3_5MoeConfig モデル)
- Qwen4ExpConfig 設定クラス: Qwen4ExpForConditionalGeneration (Qwen4ExpConfig モデル)
- ShieldGemma2Config 設定クラス: Gemma3ForConditionalGeneration (ShieldGemma2Config モデル)
- SmolVLMConfig 設定クラス: SmolVLMForConditionalGeneration (SmolVLMConfig モデル)
- Step3p7Config 設定クラス: Step3p7ForConditionalGeneration (Step3p7Config モデル)
- T5Gemma2Config 設定クラス: T5Gemma2ForConditionalGeneration (T5Gemma2Config モデル)
- UdopConfig 設定クラス: UdopForConditionalGeneration (UdopConfig モデル)
- UnlimitedOcrConfig 設定クラス: UnlimitedOcrForConditionalGeneration (UnlimitedOcrConfig モデル)
- VideoLlama3Config 設定クラス: VideoLlama3ForConditionalGeneration (VideoLlama3Config モデル)
- VideoLlavaConfig 設定クラス: VideoLlavaForConditionalGeneration (VideoLlavaConfig モデル)
- VipLlavaConfig 設定クラス: VipLlavaForConditionalGeneration (VipLlavaConfig モデル)
- VisionEncoderDecoderConfig 設定クラス: VisionEncoderDecoderModel (VisionEncoderDecoderConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(image-text-to-textモデリングヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
ライブラリのモデルクラスの一つ(画像・テキストからテキストへの変換モデリングヘッド付き)を、事前学習済みモデルからインスタンス化します。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- aria — AriaForConditionalGeneration(AriaConfig モデル)
- aya_vision — AyaVisionForConditionalGeneration(AyaVisionConfigモデル)
- blip — BlipForConditionalGeneration(BlipConfig モデル)
- blip-2 — Blip2ForConditionalGeneration(Blip2Config モデル)
- chameleon — ChameleonForConditionalGeneration(ChameleonConfig モデル)
- cohere2_vision — Cohere2VisionForConditionalGeneration(Cohere2VisionConfig モデル)
- cohere_compass — CohereCompassForConditionalGeneration(CohereCompassConfig モデル)
- cosmos3_edge — Cosmos3EdgeForConditionalGeneration(Cosmos3EdgeConfig モデル)
- cosmos3_omni — Cosmos3OmniForConditionalGeneration(Cosmos3OmniConfig モデル)
- deepseek_ocr2 — DeepseekOcr2ForConditionalGeneration (DeepseekOcr2Config モデル)
- deepseek_vl — DeepseekVLForConditionalGeneration(DeepseekVLConfig モデル)
- deepseek_vl_hybrid — DeepseekVLHybridForConditionalGeneration(DeepseekVLHybridConfig モデル)
- diffusion_gemma — DiffusionGemmaForBlockDiffusion(DiffusionGemmaConfig モデル)
- dots3_note — Dots3NoteForConditionalGeneration(Dots3NoteConfig モデル)
- emu3 — Emu3ForConditionalGeneration(Emu3Configモデル)
- ernie4_5_vl_moe — Ernie4_5_VLMoeForConditionalGeneration(Ernie4_5_VLMoeConfig モデル)
- evolla — EvollaForProteinText2Text(EvollaConfigモデル)
- exaone4_5 — Exaone4_5_ForConditionalGeneration(Exaone4_5_Config モデル)
- fast_vlm — FastVlmForConditionalGeneration(FastVlmConfigモデル)
- florence2 — Florence2ForConditionalGeneration(Florence2Config モデル)
- fuyu — FuyuForCausalLM(FuyuConfigモデル)
- gemma3 — Gemma3ForConditionalGeneration(Gemma3Config モデル)
- gemma3n — Gemma3nForConditionalGeneration(Gemma3nConfig モデル)
- gemma4 — Gemma4ForConditionalGeneration(Gemma4Config モデル)
- gemma4_unified — Gemma4UnifiedForConditionalGeneration(Gemma4UnifiedConfig モデル)
- git — GitForCausalLM(GitConfigモデル)
- glm46v — Glm46VForConditionalGeneration(Glm46VConfig モデル)
- glm4v — Glm4vForConditionalGeneration(Glm4vConfig モデル)
- glm4v_moe — Glm4vMoeForConditionalGeneration(Glm4vMoeConfig モデル)
- glm5_next — Glm5NextForConditionalGeneration(Glm5NextConfig モデル)
- glm_ocr — GlmOcrForConditionalGeneration(GlmOcrConfig モデル)
- glmga — Glm46VForConditionalGeneration(GlmgaConfig モデル)
- got_ocr2 — GotOcr2ForConditionalGeneration(GotOcr2Config モデル)
- granite4_vision — Granite4VisionForConditionalGeneration(Granite4VisionConfig モデル)
- granite_docling — GraniteDoclingForConditionalGeneration(GraniteDoclingConfigモデル)
- hunyuan_vl — HunYuanVLForConditionalGeneration (HunYuanVLConfig モデル)
- hyperclovax_vision_v2 — HyperCLOVAXVisionV2ForConditionalGeneration(HyperCLOVAXVisionV2Config モデル)
- idefics — IdeficsForVisionText2Text(IdeficsConfigモデル)
- idefics2 — Idefics2ForConditionalGeneration(Idefics2Config モデル)
- idefics3 — Idefics3ForConditionalGeneration(Idefics3Config モデル)
- inkling2 — Inkling2ForConditionalGeneration(Inkling2Configモデル)
- inkling_mm_model — InklingForConditionalGeneration(InklingConfig モデル)
- instructblip — InstructBlipForConditionalGeneration(InstructBlipConfig モデル)
- instructblipvideo — InstructBlipVideoForConditionalGeneration(InstructBlipVideoConfig モデル)
- internvl — InternVLForConditionalGeneration(InternVLConfig モデル)
- janus — JanusForConditionalGeneration(JanusConfig モデル)
- kimi_k25 — Kimi_K25ForConditionalGeneration(Kimi_K25Config モデル)
- kosmos-2 — Kosmos2ForConditionalGeneration(Kosmos2Config モデル)
- kosmos-2.5 — Kosmos2_5ForConditionalGeneration(Kosmos2_5Config モデル)
- lfm2_vl — Lfm2VlForConditionalGeneration(Lfm2VlConfig モデル)
- lfm3_vl — Lfm3VlForConditionalGeneration(Lfm3VlConfigモデル)
- lighton_ocr — LightOnOcrForConditionalGeneration(LightOnOcrConfigモデル)
- llama4 — Llama4ForConditionalGeneration(Llama4Config モデル)
- llava — LlavaForConditionalGeneration(LlavaConfigモデル)
- llava_next — LlavaNextForConditionalGeneration(LlavaNextConfig モデル)
- llava_next_video — LlavaNextVideoForConditionalGeneration(LlavaNextVideoConfig モデル)
- llava_onevision — LlavaOnevisionForConditionalGeneration(LlavaOnevisionConfig モデル)
- minicpm_ocr — MiniCPMOcrForConditionalGeneration(MiniCPMOcrConfig モデル)
- minicpmv4_6 — MiniCPMV4_6ForConditionalGeneration(MiniCPMV4_6Config モデル)
- minicpmv4_7 — MiniCPMV4_7ForConditionalGeneration(MiniCPMV4_7Config モデル)
- minimax_m3_vl — MiniMaxM3SparseForConditionalGeneration(MiniMaxM3VLConfig モデル)
- mistral3 — Mistral3ForConditionalGeneration(Mistral3Config モデル)
- mistral4 — Mistral4ForCausalLM(Mistral4Config モデル)
- mistral_large4 — MistralLarge4ForConditionalGeneration(MistralLarge4Config モデル)
- mllama — MllamaForConditionalGeneration(MllamaConfig モデル)
- molmo2 — Molmo2ForConditionalGeneration (Molmo2Config モデル)
- muse_glimmer — MuseGlimmerForConditionalGeneration (MuseGlimmerConfig モデル)
- muse_spark — MuseSparkForConditionalGeneration(MuseSparkConfigモデル)
- nemotron_h_omni — NemotronH_Omni_Reasoning_V3(NemotronH_Omni_Reasoning_V3_Config モデル)
- ovis2 — Ovis2ForConditionalGeneration(Ovis2Config モデル)
- paddleocr_vl — PaddleOCRVLForConditionalGeneration(PaddleOCRVLConfig モデル)
- paligemma — PaliGemmaForConditionalGeneration(PaliGemmaConfig モデル)
- perception_lm — PerceptionLMForConditionalGeneration(PerceptionLMConfig モデル)
- pi0 — PI0ForConditionalGeneration(PI0Configモデル)
- pix2struct — Pix2StructForConditionalGeneration(Pix2StructConfig モデル)
- pp_chart2table — GotOcr2ForConditionalGeneration(PPChart2TableConfig モデル)
- pp_formulanet — PPFormulaNetForConditionalGeneration(PPFormulaNetConfigモデル)
- qianfan_ocr — QianfanOCRForConditionalGeneration(QianfanOCRConfig モデル)
- qwen2_5_omni_thinker — Qwen2_5OmniThinkerForConditionalGeneration (Qwen2_5OmniThinkerConfig モデル)
- qwen2_5_vl — Qwen2_5_VLForConditionalGeneration(Qwen2_5_VLConfig モデル)
- qwen2_vl — Qwen2VLForConditionalGeneration(Qwen2VLConfigモデル)
- qwen3_5 — Qwen3_5ForConditionalGeneration(Qwen3_5Config モデル)
- qwen3_5_moe — Qwen3_5MoeForConditionalGeneration(Qwen3_5MoeConfig モデル)
- qwen3_omni_moe_thinker — Qwen3OmniMoeThinkerForConditionalGeneration(Qwen3OmniMoeThinkerConfig モデル)
- qwen3_vl — Qwen3VLForConditionalGeneration(Qwen3VLConfig モデル)
- qwen3_vl_moe — Qwen3VLMoeForConditionalGeneration(Qwen3VLMoeConfig モデル)
- qwen4_exp — Qwen4ExpForConditionalGeneration(Qwen4ExpConfig モデル)
- shieldgemma2 — Gemma3ForConditionalGeneration(ShieldGemma2Config モデル)
- smolvlm — SmolVLMForConditionalGeneration(SmolVLMConfig モデル)
- step3p7 — Step3p7ForConditionalGeneration(Step3p7Config モデル)
- t5gemma2 — T5Gemma2ForConditionalGeneration(T5Gemma2Config モデル)
- udop — UdopForConditionalGeneration (UdopConfig モデル)
- unlimited_ocr — UnlimitedOcrForConditionalGeneration(UnlimitedOcrConfigモデル)
- video_llama_3 — VideoLlama3ForConditionalGeneration (VideoLlama3Config モデル)
- video_llava — VideoLlavaForConditionalGeneration(VideoLlavaConfigモデル)
- vipllava — VipLlavaForConditionalGeneration(VipLlavaConfig モデル)
- vision-encoder-decoder — VisionEncoderDecoderModel(VisionEncoderDecoderConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForImageTextToText
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForImageTextToText.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForImageTextToText.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
TrueAutoModelForDecision
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、決定ヘッド付きのライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- Gemma4Config 設定クラス: JevForDecision (Gemma4Config モデル)
- JevConfig 設定クラス: JevForDecision (JevConfig モデル)
- Qwen2Config 設定クラス: JevForDecision (Qwen2Config モデル)
- Qwen3_5Config 設定クラス: JevForDecision (Qwen3_5Config モデル)
- Qwen3_5MoeConfig 設定クラス: JevForDecision (Qwen3_5MoeConfig モデル)
- Qwen3_5TextConfig 設定クラス: JevForDecision (Qwen3_5TextConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(決定ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
…
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- gemma4 — JevForDecision(Gemma4Configモデル)
- jev — JevForDecision(JevConfigモデル)
- qwen2 — JevForDecision(Qwen2Configモデル)
- qwen3_5 — JevForDecision(Qwen3_5Config モデル)
- qwen3_5_moe — JevForDecision(Qwen3_5MoeConfig モデル)
- qwen3_5_text — JevForDecision(Qwen3_5TextConfig モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForDecision
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForDecision.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForDecision.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True時系列
AutoModelForTimeSeriesPrediction
これは汎用的なモデルクラスであり、from_pretrained()クラスメソッドまたはfrom_config()クラスメソッドで生成された場合、タイムシリーズ予測ヘッドを備えたライブラリのモデルクラスのいずれかとしてインスタンス化されます。
このクラスは__init__()を直接使用してインスタンス化することはできません(エラーが発生します)。
from_config
< source >( **kwargs )
パラメータ
- config (PreTrainedConfig) —
インスタンス化されるモデルクラスは、設定クラスに基づいて選択されます:
- TimesFm2_5Config 設定クラス: TimesFm2_5ModelForPrediction (TimesFm2_5Config モデル)
- TimesFmConfig 設定クラス: TimesFmModelForPrediction (TimesFmConfig モデル)
- attn_implementation (
str, 省略可) — モデルで使用されるアテンションの実装(関連する場合)。"eager"(アテンションの手動実装)"sdpa"(F.scaled_dot_product_attentionを使用)"flash_attention_2"(Dao-AILab/flash-attentionを使用)"flash_attention_3"(Dao-AILab/flash-attention/hopperを使用)"flash_attention_4"(Dao-AILab/flash-attention/flash_attn/cuteを使用)。 デフォルトでは、利用可能な場合、torch>=2.1.1 では SDPA が使用されます。それ以外の場合は、デフォルトで手動の"eager"実装が使用されます。
設定から、ライブラリのモデルクラスの一つ(時系列予測ヘッド付き)をインスタンス化します。
注: 設定ファイルからモデルを読み込んでも、モデルの重みは読み込まれません。影響するのはモデルの設定のみです。モデルの重みを読み込むには、from_pretrained()を使用してください。
from_pretrained
< source >( *model_args**kwargs )
パラメータ
- pretrained_model_name_or_path (
strまたはos.PathLike) — 以下のいずれかです:- huggingface.co のモデルリポジトリにホストされている事前学習済みモデルの model id を示す文字列。
- save_pretrained() を使用して保存されたモデルの重みを含むディレクトリへのパス。例:
./my_model_directory/
- model_args(追加の位置引数、省略可)—
基盤となるモデルの
__init__()メソッドに渡されます。 - config (PreTrainedConfig, 省略可) —
自動的に読み込まれる設定の代わりに使用するモデルの設定。以下の条件を満たす場合、設定は自動的に読み込まれます:
- モデルは、ライブラリによって提供されるモデルです(事前学習済みモデルのmodel id文字列で読み込まれます)。
- モデルはsave_pretrained()を使用して保存されており、保存ディレクトリを指定することで再読み込みされます。
- モデルは、ローカルディレクトリを
pretrained_model_name_or_pathとして指定して読み込まれ、そのディレクトリ内にconfig.jsonという名前の設定JSONファイルが見つかった場合です。
- state_dict (dict[str, torch.Tensor], 省略可) —
保存された重みファイルから読み込まれる state dictionary の代わりに使用する state dictionary です。事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合にこのオプションを使用できます。ただし、この場合は save_pretrained() と from_pretrained() を使用した方が簡単ではないかを確認することをお勧めします。
このオプションは、事前学習済みの設定からモデルを作成しつつ、独自の重みをロードしたい場合に使用できます。ただし、この場合は、save_pretrained()とfrom_pretrained()を使用する方がよりシンプルな方法ではないかを確認することをお勧めします。
- cache_dir (
strまたはos.PathLike、省略可) — 標準のキャッシュを使用しない場合、ダウンロードした事前学習済みモデルの設定をキャッシュするディレクトリのパス。 - force_download (
bool、任意、デフォルトはFalse) — モデルの重みと設定ファイルを(再)ダウンロードを強制するかどうか。キャッシュされたバージョンが存在する場合でも、それを上書きします。 - proxies (
dict[str, str]、省略可) — プロトコルまたはエンドポイントごとに使用するプロキシサーバーの辞書です。例:{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}。各リクエストでこれらのプロキシが使用されます。 - output_loading_info(
bool, optional, デフォルトはFalse) — 欠落キー、予期しないキー、エラーメッセージを含む辞書を返すかどうかを指定します。 - local_files_only(
bool, 任意、デフォルトはFalse) — ローカルファイルのみを対象とするかどうか(モデルのダウンロードを試みないなど)。 - revision (
str、任意、デフォルトは"main") — 使用する特定のモデルのバージョンです。huggingface.co 上でモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、ブランチ名、タグ名、コミット ID のいずれかを指定できます。したがって、revisionには git で許可される任意の識別子を指定できます。 - trust_remote_code (
bool、省略可、デフォルトはFalse) — Hub 上で独自の modeling ファイルによって定義されたカスタムモデルを許可するかどうか。このオプションは、信頼できるリポジトリで、かつコードを確認済みの場合にのみTrueに設定してください。そうしないと、Hub に存在するコードがローカルマシン上で実行されてしまうためです。 - code_revision (
str、任意、デフォルトは"main") — モデルの他の部分とは異なるリポジトリにコードが存在する場合、Hub 上のコードに使用する特定のリビジョンです。huggingface.co ではモデルやその他の成果物を保存するために git ベースのシステムを使用しているため、revisionには git で許可される任意の識別子を使用できます。具体的には、ブランチ名、タグ名、またはコミット ID を指定できます。 - kwargs (追加のキーワード引数、 省略可能) —
設定オブジェクト(読み込み後)の更新やモデルの初期化(例:
output_attentions=True). が提供されるかどうかによって動作が異なります。configが提供される場合、または自動的に読み込まれる場合:configで設定が提供されている場合、**kwargsは、基盤となるモデルの__init__メソッドに直接渡されます(設定への関連するすべての更新がすでに完了しているものと仮定しています)。- 設定が提供されない場合、
kwargsはまず設定クラスの初期化関数 (from_pretrained()) に渡されます。kwargsの各キーのうち、設定属性に対応するものは、指定されたkwargsの値でその属性を上書きするために使用されます。設定属性に対応しない残りのキーは、基盤となるモデルの__init__関数に渡されます。
ライブラリのモデルクラスの一つ(時系列予測ヘッド付き)を事前学習済みモデルからインスタンス化します。
インスタンス化されるモデルクラスは、設定オブジェクトのmodel_typeプロパティ(引数として渡されたもの、または可能な場合はpretrained_model_name_or_pathから読み込まれたもの)に基づいて選択されます。このプロパティが存在しない場合は、pretrained_model_name_or_pathに対するパターンマッチングにフォールバックして選択されます。
- timesfm — TimesFmModelForPrediction (TimesFmConfig モデル)
- timesfm2_5 — TimesFm2_5ModelForPrediction(TimesFm2_5Config モデル)
モデルはデフォルトでmodel.eval()によって評価モードに設定されています(例えば、dropoutモジュールは無効化されます)。モデルを学習させるには、まずmodel.train()でトレーニングモードに戻す必要があります。
例:
>>> from transformers import AutoConfig, AutoModelForTimeSeriesPrediction
>>> # Download model and configuration from huggingface.co and cache.
>>> model = AutoModelForTimeSeriesPrediction.from_pretrained("google-bert/bert-base-cased")
>>> # Update configuration during loading
>>> model = AutoModelForTimeSeriesPrediction.from_pretrained("google-bert/bert-base-cased", output_attentions=True)
>>> model.config.output_attentions
True