[Misc] Remove OLMo2 config copy (#17066)
Signed-off-by: Isotr0py <2037008807@qq.com>
This commit is contained in:
@@ -28,6 +28,7 @@ from typing import Iterable, Optional, Tuple, Union
|
||||
|
||||
import torch
|
||||
from torch import nn
|
||||
from transformers import Olmo2Config
|
||||
|
||||
from vllm.attention import Attention
|
||||
from vllm.config import VllmConfig
|
||||
@@ -51,7 +52,6 @@ from vllm.model_executor.models.utils import (
|
||||
make_layers, maybe_prefix)
|
||||
from vllm.model_executor.sampling_metadata import SamplingMetadata
|
||||
from vllm.sequence import IntermediateTensors
|
||||
from vllm.transformers_utils.configs.olmo2 import Olmo2Config
|
||||
|
||||
|
||||
class Olmo2Attention(nn.Module):
|
||||
|
||||
Reference in New Issue
Block a user