Unverified Commit cd1245a1 authored by maang's avatar maang Committed by GitHub
Browse files

[Cleanup] Remove redundant `decoder_layer_type` assignment in `Qwen2` (#31760)


Signed-off-by: default avatarmaang <maang_h@163.com>
parent ffec8154
...@@ -389,8 +389,6 @@ class Qwen2Model(nn.Module): ...@@ -389,8 +389,6 @@ class Qwen2Model(nn.Module):
else: else:
self.embed_tokens = PPMissingLayer() self.embed_tokens = PPMissingLayer()
# Use the provided decoder layer type or default to Qwen2DecoderLayer
decoder_layer_type = decoder_layer_type or Qwen2DecoderLayer
self.start_layer, self.end_layer, self.layers = make_layers( self.start_layer, self.end_layer, self.layers = make_layers(
config.num_hidden_layers, config.num_hidden_layers,
lambda prefix: decoder_layer_type( lambda prefix: decoder_layer_type(
......
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