File tree 2 files changed +3
-2
lines changed 2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -73,8 +73,9 @@ def nanotron(
73
73
skip_null_keys = True ,
74
74
)
75
75
76
- # We are getting a type error, because the get_config_from_file is not correctly typed,
76
+ # Load lighteval config
77
77
lighteval_config : LightEvalConfig = get_config_from_file (lighteval_config_path , config_class = LightEvalConfig ) # type: ignore
78
+
78
79
nanotron_config = FullNanotronConfig (lighteval_config , model_config )
79
80
80
81
evaluation_tracker = EvaluationTracker (
Original file line number Diff line number Diff line change @@ -187,7 +187,7 @@ def _init_parallelism_manager(self):
187
187
def _init_model (self , model_config , model ):
188
188
logger .info ("--- LOADING MODEL ---" )
189
189
if model_config is not None :
190
- if self .parallel_context :
190
+ if self .parallel_context :
191
191
return NanotronLightevalModel (
192
192
checkpoint_path = os .path .dirname (self .pipeline_parameters .nanotron_checkpoint_path )
193
193
if self .pipeline_parameters .nanotron_checkpoint_path
You can’t perform that action at this time.
0 commit comments