Fix backbone weights loading bug.

SamanehSaadat · Apr 11, 2024 · 90f6679 · 90f6679
1 parent 5d9dd82
commit 90f6679
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_nlp/models/backbone.py b/keras_nlp/models/backbone.py
@@ -209,7 +209,7 @@ class like `keras_nlp.models.Backbone.from_preset()`, or from
         backbone = load_serialized_object(preset, CONFIG_FILE)
         if load_weights:
             jax_memory_cleanup(backbone)
-            backbone.load_weights(get_file(preset, CONFIG_FILE))
+            backbone.load_weights(get_file(preset, MODEL_WEIGHTS_FILE))
 
         return backbone