search

yuanzhi-zhu / prolific_dreamer2d

Unofficial implementation of 2D ProlificDreamer
130 stars 6 forks source link

What's the differences between VSD command line multiple particles and VSD command line? #9

Closed VLadImirluren closed 11 months ago

VLadImirluren commented 11 months ago

thanks for your greate implementation!

VLadImirluren commented 11 months ago

there is a bug in VSD command line multiple particles: Traceback (most recent call last): File "prolific_dreamer2d.py", line 539, in main() File "prolific_dreamer2d.py", line 489, in main save_image((image/2+0.5).clamp(0, 1), f'{args.work_dir}/{image_name}_image_step{step}_t{t.item()}.png') UnboundLocalError: local variable 'image' referenced before assignment

yuanzhi-zhu commented 11 months ago

Hi @VLadImirluren thank you for your interest and bug report

  1. in VSD command line, we use one particle by default.
  2. A quick fix has been made.
VLadImirluren commented 11 months ago

thanks! @yuanzhi-zhu