neuralmagic / sparsezoo

Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes
Apache License 2.0
371 stars 25 forks source link

[BugFix] Fix Serialization of Computed Properties in BaseModel #485

Closed rahul-tuli closed 7 months ago

rahul-tuli commented 7 months ago

Description:

A bug was identified where computed properties in the base model were not serialized correctly due to an overridden dict() method. The method was initially designed to inject properties, but since the introduction of model_dump as a naming convention, it no longer executed the necessary code when yaml() was called.

Fixes Included:

Test Plan: