Tonic commited on
Commit
5fe45a6
·
verified ·
1 Parent(s): 93eabe8

add error handling for monitoring

Browse files
Files changed (1) hide show
  1. model.py +11 -1
model.py CHANGED
@@ -172,7 +172,17 @@ class SmolLM3Model:
172
  # Override with kwargs
173
  training_args.update(kwargs)
174
 
175
- return TrainingArguments(**training_args)
 
 
 
 
 
 
 
 
 
 
176
 
177
  def save_pretrained(self, path: str):
178
  """Save model and tokenizer"""
 
172
  # Override with kwargs
173
  training_args.update(kwargs)
174
 
175
+ # Debug: Check for any boolean values that might be causing issues
176
+ for key, value in training_args.items():
177
+ if isinstance(value, bool):
178
+ logger.info(f"Boolean argument: {key} = {value}")
179
+
180
+ try:
181
+ return TrainingArguments(**training_args)
182
+ except Exception as e:
183
+ logger.error(f"Failed to create TrainingArguments: {e}")
184
+ logger.error(f"Training arguments: {training_args}")
185
+ raise
186
 
187
  def save_pretrained(self, path: str):
188
  """Save model and tokenizer"""