keras/src/layers/preprocessing/index_lookup.py/IndexLookup/load_own_variables
class IndexLookup:    def load_own_variables(self, store):
        if self.output_mode == "tf_idf":
            self.idf_weights.assign(store["idf_weights"])
            self.idf_weights_const = self.idf_weights.value()