xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-09-09 01:17:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc]: Typo fix: Renaming classes (casualLM -> causalLM) (#9801)

Browse Source 
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
This commit is contained in:
yannicks1 2024-10-29 18:39:20 +01:00 committed by GitHub
parent 0f43387157
commit 0ce7798f44
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vllm/model_executor/model_loader/neuron.py
View File

@ -37,7 +37,7 @@ _NEURON_SUPPORTED_MODELS: Dict[str, Tuple[str, str, str]] = {
} }
class NeuronCasualLM(nn.Module): class NeuronCausalLM(nn.Module):
def __init__(self, def __init__(self,
config: PretrainedConfig, config: PretrainedConfig,
@ -184,7 +184,7 @@ def get_neuron_model(model_config: ModelConfig,
scheduler_config: SchedulerConfig) -> nn.Module: scheduler_config: SchedulerConfig) -> nn.Module:
# Create a model instance. # Create a model instance.
model = NeuronCasualLM( model = NeuronCausalLM(
model_config.hf_config, model_config.hf_config,
_is_neuron_on_device_sampling_disabled(model_config)) _is_neuron_on_device_sampling_disabled(model_config))

4
vllm/model_executor/model_loader/openvino.py
View File

@ -95,7 +95,7 @@ def _require_model_export(model_id, revision=None, subfolder=None):
return True return True
class OpenVINOCasualLM(nn.Module): class OpenVINOCausalLM(nn.Module):
def __init__( def __init__(
self, self,
@ -199,5 +199,5 @@ def get_model(
"be added in the future. If this is important to you, " "be added in the future. If this is important to you, "
"please open an issue on github.") "please open an issue on github.")
return OpenVINOCasualLM(ov_core, model_config, device_config, return OpenVINOCausalLM(ov_core, model_config, device_config,
kv_cache_dtype) kv_cache_dtype)