Skip to content

Commit

Permalink
fix: switching to new evaluate function
Browse files Browse the repository at this point in the history
  • Loading branch information
timurcarstensen committed Feb 13, 2025
1 parent 0851663 commit 0aab4d9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion whittle/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
from litgpt.deploy.serve import run_server as serve_fn
from litgpt.scripts.download import download_from_hub as download_fn

from whittle.eval.utils import convert_and_evaluate as evaluate_fn
from whittle.evaluate_network import setup as evaluate_fn
from whittle.finetune import setup as finetune_fn
from whittle.pretrain_super_network import setup as pretrain_fn
from whittle.search_sub_networks import setup as search_fn
Expand Down

0 comments on commit 0aab4d9

Please sign in to comment.