search

giuvecchio / matfuse-sd

MatFuse: Controllable Material Generation with Diffusion Models (CVPR2024)
https://gvecchio.com/matfuse/
MIT License
31 stars 2 forks source link

Training crashes on step 2000 #5

Closed Night1099 closed 4 months ago

Night1099 commented 4 months ago

I get this error when script reaches step 2000 I assume it's a img log error

TypeError: VQModelMulti.forward() missing 1 required positional argument: 'return_pred_indices'
rensortino commented 4 months ago

There was a default value not set in the VQModelMulti class, fixed in https://github.com/giuvecchio/matfuse-sd/commit/59e867de7d59b732f977bc4dac65983366ff3d09

Night1099 commented 4 months ago

Thanks!