Skip to content

[BUG] Can not load a Prunamodel from hub because smash_config does not contain the new family of algo #116

@llcnt

Description

@llcnt

Describe the bug

I can not load this model from hub because smash_config does not contain the new family of algo FACTORIZER.

What I did

I have a PR that introduces the new family FACTORIZER: #75
But unit test are failing for the function test_pruna_model_from_hub , because the smash_config was uploaded on the hub without this newly created family of algo.
I guess this error will come back in the future if a new family is again created, or if the smash_config is refactored, and if a user wants to load an "old" model ;)

Paste the command(s) you ran and the output.
If there was a crash, please include the traceback here.

Expected behavior

Screenshots

Environment

  • pruna version: 0.2.4
  • python version: 3.10
  • Operating System: ubuntu

Additional context

Metadata

Metadata

Labels

bugSomething isn't working

Type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions