"tests/vscode:/vscode.git/clone" did not exist on "c40692bf9adc77cde0c802c6c586b7cf143c39c2"
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]] = {
}
class NeuronCasualLM(nn.Module):
class NeuronCausalLM(nn.Module):
def __init__(self,
config: PretrainedConfig,
......@@ -184,7 +184,7 @@ def get_neuron_model(model_config: ModelConfig,
scheduler_config: SchedulerConfig) -> nn.Module:
# Create a model instance.
model = NeuronCasualLM(
model = NeuronCausalLM(
model_config.hf_config,
_is_neuron_on_device_sampling_disabled(model_config))
......
......@@ -95,7 +95,7 @@ def _require_model_export(model_id, revision=None, subfolder=None):
return True
class OpenVINOCasualLM(nn.Module):
class OpenVINOCausalLM(nn.Module):
def __init__(
self,
......@@ -199,5 +199,5 @@ def get_model(
"be added in the future. If this is important to you, "
"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)
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