ModuleNotFoundError: No module named 'gdf'
vardhanam opened this issue · comments
python3 train/train_c_lora.py configs/training/finetune_c_3b_lora.yaml
running the above is giving me an error
File "/home/vardh/fine_tune_stable_cascade/StableCascade/train/train_c_lora.py", line 12, in <module>
from gdf import GDF, EpsilonTarget, CosineSchedule
ModuleNotFoundError: No module named 'gdf'
I can't find this module anywhere on the web. Please help
pip install git+https://github.com/WARP-AI/gdf -U
see here #10
Adding sys.path.append(os.getcwd())
seems work to me.
So the module import should looks like
sys.path.append(os.getcwd())
from gdf import GDF, EpsilonTarget, CosineSchedule