Fix gemma 3 launch with transformers: the error: AttributeError: 'TransformersForCausalLM' object has no attribute 'tp_size' (#9614)

2025-09-29 06:18:29 +00:00
parent 24f7cb1ece
commit 71959545df
1 changed files with 0 additions and 2 deletions
@@ -20,7 +20,6 @@ import torch.nn.functional as F
 from torch import nn
 from transformers import (
    ROPE_INIT_FUNCTIONS,
-    AutoModel,
    Gemma3TextConfig,
    PretrainedConfig,
    PreTrainedModel,
@@ -761,4 +760,3 @@ class Gemma3ForCausalLM(PreTrainedModel):


 EntryClass = Gemma3ForCausalLM
-AutoModel.register(Gemma3TextConfig, Gemma3ForCausalLM, exist_ok=True)