ModuleNotFoundError: No module named 'gdf'

[vardhanam]

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

[SirTrippsalot]

pip install git+https://github.com/WARP-AI/gdf -U

[geroldmeisinger]

see here https://github.com/Stability-AI/StableCascade/issues/10

[tikhonlavrev]

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