Skip to content

Commit 503ed97

Browse files
committed
Merge branch 'nanotron-fix' into upd-nanotron
# Conflicts: # src/lighteval/config/lighteval_config.py # src/lighteval/main_nanotron.py # src/lighteval/models/nanotron/nanotron_model.py
2 parents 826bd83 + 03c1195 commit 503ed97

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

src/lighteval/main_nanotron.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,8 +73,9 @@ def nanotron(
7373
skip_null_keys=True,
7474
)
7575

76-
# We are getting a type error, because the get_config_from_file is not correctly typed,
76+
# Load lighteval config
7777
lighteval_config: LightEvalConfig = get_config_from_file(lighteval_config_path, config_class=LightEvalConfig) # type: ignore
78+
7879
nanotron_config = FullNanotronConfig(lighteval_config, model_config)
7980

8081
evaluation_tracker = EvaluationTracker(

src/lighteval/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ def _init_parallelism_manager(self):
187187
def _init_model(self, model_config, model):
188188
logger.info("--- LOADING MODEL ---")
189189
if model_config is not None:
190-
if self.parallel_context:
190+
if self.parallel_context:
191191
return NanotronLightevalModel(
192192
checkpoint_path=os.path.dirname(self.pipeline_parameters.nanotron_checkpoint_path)
193193
if self.pipeline_parameters.nanotron_checkpoint_path

0 commit comments

Comments
 (0)