Unverified Commit 71959545 authored by Vincent Zhong's avatar Vincent Zhong Committed by GitHub
Browse files

Fix gemma 3 launch with `transformers:` the error: `AttributeError:...

Fix gemma 3 launch with `transformers:` the error: `AttributeError: 'TransformersForCausalLM' object has no attribute 'tp_size'` (#9614)
parent 24f7cb1e
......@@ -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)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment