Logical bug of `self.img_embed`
hubert0527 opened this issue · 1 comments
hubert0527 commented
The persistent attribute self.img_embed
is assigned to the current image latent here:
GeoWizard/geowizard/models/geowizard_pipeline.py
Lines 248 to 249 in 0a60193
It is understandable to save AE computation, but the variable is not reset after the depth/normal ensemble. Thus, making a second call to the pipeline inference will result in the wrong input condition.
A simple (but not so clean) fix is to assign self.img_embed=None
before return:
GeoWizard/geowizard/models/geowizard_pipeline.py
Lines 199 to 201 in 0a60193
fuxiao0719 commented
Thanks, we have fixed it.