Handle cached model missing file exception.

SamanehSaadat · Apr 15, 2024 · 55396ad · 55396ad
1 parent 38243e6
commit 55396ad
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/keras_nlp/utils/preset_utils.py b/keras_nlp/utils/preset_utils.py
@@ -120,6 +120,14 @@ def get_file(preset, path):
                 )
             else:
                 raise ValueError(message)
+        except ValueError as e:
+            message = str(e)
+            if message.find("is not present in the model files"):
+                raise FileNotFoundError(
+                    f"`{path}` doesn't exist in preset directory `{preset}`."
+                )
+            else:
+                raise ValueError(message)
 
     elif preset.startswith(GS_PREFIX):
         url = os.path.join(preset, path)