FartyPants / Training_PRO

Traing PRO extension for oobabooga WebUI - recent dev version
MIT License
46 stars 2 forks source link

Failed to load the extension #4

Closed krslim closed 1 year ago

krslim commented 1 year ago

Thanks for your work! My oogabooga version is working, but I'm getting the following error with the latest code from this repo:

2023-10-18 19:30:53 ERROR:Failed to load the extension "Training_PRO". Traceback (most recent call last): File "F:\AI\Oobabooga\text-generation-webui\modules\extensions.py", line 36, in load_extensions exec(f"import extensions.{name}.script") File "", line 1, in File "F:\AI\Oobabooga\text-generation-webui\extensions\Training_PRO\script.py", line 22, in from .custom_scheduler import FPSchedulerTrainer, custom_scheduler_global_update, custom_scheduler_global_setup, custom_scheduler_global_shuffle ImportError: cannot import name 'custom_scheduler_global_update' from 'extensions.Training_PRO.custom_scheduler' (F:\AI\Oobabooga\text-generation-webui\extensions\Training_PRO\custom_scheduler.py)

Does custom_scheduler.py need to be updated to reflect the recent changes with script.py?

SrVill commented 1 year ago

A similar problem. I went back to commit ac4e782. Then there is no problem

FartyPants commented 1 year ago

You need to get all the files from here. The project here is the latest project I'm running on my computer.

krslim commented 1 year ago

I'm looking forward to experimenting with the mistral models, thanks again!