EternalEvan / FlowIE

This repository contains the official implementation of "FlowIE: Efficient Image Enhancement via Rectified Flow"
MIT License
76 stars 2 forks source link

Names of model weights and bias don't match those in the ckpt #8

Closed AnnyShen55 closed 3 months ago

AnnyShen55 commented 3 months ago

Currently, except the proj layers, other layers in model.diffusionmodel are not of type nn.linear. As a result, the lora layers cannot generate from those layers, which lead to mismatches between the names of the model weights and bias and those in the checkpoint files. I guess there are some updates in your model, can you upload the new model file(s)? Thank you!