Update keras/src/layers/layer.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: divyashreepathihalli

keras/src/layers/layer.py

@@ -1440,9 +1440,8 @@ def _load_quantized_variables(self, store):
         # Load non-quantized variables
         all_vars = self._trainable_variables + self._non_trainable_variables
         non_quantized_vars = [
-            v for v in all_vars if v not in quantized_vars and v.trainable
+            v for v in all_vars if v not in quantized_vars
         ]
-        for i, v in enumerate(non_quantized_vars):
             v.assign(store[f"{i}"])
 
     def _track_variable(self, variable):