locuslab / wanda

A simple and effective LLM pruning approach.
https://arxiv.org/abs/2306.11695
MIT License
615 stars 79 forks source link

AttributeError: module 'lm_eval.tasks' has no attribute 'ALL_TASKS' #62

Open Shinning-Zhou opened 1 month ago

Shinning-Zhou commented 1 month ago

met

Traceback (most recent call last):
  File "/mnt/sharedata/hdd/zhouxn/wanda/main.py", line 112, in <module>
    main()
  File "/mnt/sharedata/hdd/zhouxn/wanda/main.py", line 102, in main
    results = eval_zero_shot(args.model, model, tokenizer, task_list, num_shot, accelerate)
  File "/mnt/sharedata/hdd/zhouxn/wanda/lib/eval.py", line 142, in eval_zero_shot
    task_names = pattern_match(task_list, tasks.ALL_TASKS)
AttributeError: module 'lm_eval.tasks' has no attribute 'ALL_TASKS'

when running eval_zero_shot. Can you share the version of lm_eval libaray or tell me how to fix this?

Logan-007L commented 2 weeks ago

Hello, have you solved it yet? Can you tell me how to solve it? thank you!