cpu for `presample_noise.py`

jzhoulab / ddsm

Dirichlet Diffusion Score Model for Biological Sequence Generation.

Other

45 stars 8 forks source link

cpu for `presample_noise.py` #4

Closed guillaumehu closed 10 months ago

guillaumehu commented 11 months ago

Hello, thanks for the great code! I just did a small modification to automatically set the device when running presample_noise.py. It was only working on gpu.