Unverified Commit 0ce7798f authored by yannicks1's avatar yannicks1 Committed by GitHub
Browse files

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


Signed-off-by: default avatarYannick Schnider <Yannick.Schnider1@ibm.com>
parent 0f433871
...@@ -37,7 +37,7 @@ _NEURON_SUPPORTED_MODELS: Dict[str, Tuple[str, str, str]] = { ...@@ -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, ...@@ -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))
......
...@@ -95,7 +95,7 @@ def _require_model_export(model_id, revision=None, subfolder=None): ...@@ -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( ...@@ -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)
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