diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py index f1267f15c..8745022a8 100644 --- a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py +++ b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py @@ -380,7 +380,7 @@ def use_post_layer_norm(self) -> bool: return False -class LibriSpeechConformerGeluWorkload(LibriSpeechConformerWorkload): +class LibriSpeechConformerGELUWorkload(LibriSpeechConformerWorkload): @property def use_gelu(self) -> bool: diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py index 18853d19c..f33d75cd4 100644 --- a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py +++ b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py @@ -346,7 +346,7 @@ def use_post_layer_norm(self) -> bool: return False -class LibriSpeechConformerGeluWorkload(LibriSpeechConformerWorkload): +class LibriSpeechConformerGELUWorkload(LibriSpeechConformerWorkload): @property def use_gelu(self) -> bool: diff --git a/algorithmic_efficiency/workloads/workloads.py b/algorithmic_efficiency/workloads/workloads.py index a972700d3..4231ea9f5 100644 --- a/algorithmic_efficiency/workloads/workloads.py +++ b/algorithmic_efficiency/workloads/workloads.py @@ -58,7 +58,7 @@ }, 'librispeech_conformer_gelu': { 'workload_path': 'librispeech_conformer/librispeech', - 'workload_class_name': 'LibriSpeechConformerGeluWorkload', + 'workload_class_name': 'LibriSpeechConformerGELUWorkload', }, 'librispeech_deepspeech': { 'workload_path': 'librispeech_deepspeech/librispeech', diff --git a/tests/modeldiffs/librispeech_conformer_gelu/compare.py b/tests/modeldiffs/librispeech_conformer_gelu/compare.py index 892040b57..ad695851b 100644 --- a/tests/modeldiffs/librispeech_conformer_gelu/compare.py +++ b/tests/modeldiffs/librispeech_conformer_gelu/compare.py @@ -8,9 +8,9 @@ from algorithmic_efficiency import spec from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_jax.workload import \ - LibriSpeechConformerGeluWorkload as JaxWorkload + LibriSpeechConformerGELUWorkload as JaxWorkload from algorithmic_efficiency.workloads.librispeech_conformer.librispeech_pytorch.workload import \ - LibriSpeechConformerGeluWorkload as PytWorkload + LibriSpeechConformerGELUWorkload as PytWorkload from tests.modeldiffs.diff import out_diff