From 68469d70d2298e3ff52e72e0e4130b2e91c6052b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=9C=D0=B0=D0=BB=D0=B0=D1=85=D0=BE=D0=B2=20=D0=90=D0=BB?= =?UTF-8?q?=D0=B5=D0=BA=D1=81=D0=B5=D0=B9=20=D0=9F=D0=B0=D0=B2=D0=BB=D0=BE?= =?UTF-8?q?=D0=B2=D0=B8=D1=87?= Date: Mon, 18 Nov 2024 09:51:24 +0000 Subject: [PATCH] fix --- turbo_alignment/common/tf/loaders/model/model.py | 1 - turbo_alignment/settings/model.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/turbo_alignment/common/tf/loaders/model/model.py b/turbo_alignment/common/tf/loaders/model/model.py index 94fa96f..9dc407e 100755 --- a/turbo_alignment/common/tf/loaders/model/model.py +++ b/turbo_alignment/common/tf/loaders/model/model.py @@ -119,4 +119,3 @@ def load_model( model.base_model.model.score.weight.requires_grad = True return model - diff --git a/turbo_alignment/settings/model.py b/turbo_alignment/settings/model.py index c57a157..72cd887 100755 --- a/turbo_alignment/settings/model.py +++ b/turbo_alignment/settings/model.py @@ -23,7 +23,7 @@ class LigerKernelSettings(ExtraFieldsNotAllowedBaseModel): use_rms_norm: bool = False @model_validator(mode='after') - def correct_cross_entopy_kernels(self) -> 'LigerKernelSettings': + def check_cross_entopy_kernels(self) -> 'LigerKernelSettings': if self.use_fused_linear_cross_entropy and self.use_cross_entropy: raise ValueError( 'You cannot use both FusedLinearCrossEntropy and CrossEntropy kernels. '