Spaces:
Sleeping
Sleeping
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
|
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
|
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
|