non2013 commited on
Commit
1084a38
·
1 Parent(s): d524771

rename model

Browse files
DeBERTaV3.py CHANGED
@@ -87,11 +87,11 @@ class ModelLoader:
87
  metrics=[accuracy, F1Score()]).to_fp16()
88
  try:
89
  # First attempt: Try loading with weights_only=True
90
- self.learn.load('fastai_QIQC-deberta-v3-unclean', strict=False, weights_only=True)
91
  except Exception as e:
92
  print(f"Warning: Could not load with weights_only=True. Falling back to default loading. Error: {e}")
93
  # Second attempt: Fall back to regular loading if the first attempt fails
94
- self.learn.load('fastai_QIQC-deberta-v3-unclean', strict=False)
95
 
96
  def get_learner(self):
97
  return self.learn
 
87
  metrics=[accuracy, F1Score()]).to_fp16()
88
  try:
89
  # First attempt: Try loading with weights_only=True
90
+ self.learn.load('fastai_QIQC-deberta-v3', strict=False, weights_only=True)
91
  except Exception as e:
92
  print(f"Warning: Could not load with weights_only=True. Falling back to default loading. Error: {e}")
93
  # Second attempt: Fall back to regular loading if the first attempt fails
94
+ self.learn.load('fastai_QIQC-deberta-v3', strict=False)
95
 
96
  def get_learner(self):
97
  return self.learn
models/{fastai_QIQC-deberta-v3-unclean.pth → fastai_QIQC-deberta-v3.pth} RENAMED
File without changes