We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 89ab0f1 commit 920ae45Copy full SHA for 920ae45
src/Models/Pretrained/Qwen2PreTrainedModel.php
@@ -13,7 +13,7 @@
13
/**
14
* The bare Qwen2 Model outputting raw hidden-states without any specific head on top.
15
*/
16
-class Qwen2PreTrainedModel extends PreTrainedModel
+class Qwen2PreTrainedModel extends PretrainedModel
17
{
18
protected int $numHeads;
19
protected int $numLayers;
@@ -36,4 +36,4 @@ public function __construct(
36
$this->numLayers = $this->config['num_hidden_layers'];
37
$this->dimKv = $this->config['hidden_size'] / $this->config['num_attention_heads'];
38
}
39
-}
+}
0 commit comments