Zeju1997 / oft

Official implementation of "Controlling Text-to-Image Diffusion by Orthogonal Finetuning".
https://oft.wyliu.com/
MIT License
280 stars 14 forks source link

Incorrect import in train_dreambooth.py #11

Closed masinov closed 11 months ago

masinov commented 1 year ago

The script train_dreambooth.py contains the following erroneous import:

from diffusers.utils import MHE_db as MHE

MHE isn't implemented in the diffusers library, and it is actually your own loss function. The actual MHE_db is located in oft_utils, right inside oft-db.

Changing that line to:

from oft_utils import MHE_db as MHE

as it is done in some other of your scripts solves the issue.

YuxuanSnow commented 11 months ago

Hi, many thanks for pointing this out!