search

FreeStyleFreeLunch / FreeStyle

FreeStyle : Free Lunch for Text-guided Style Transfer using Diffusion Models
109 stars 7 forks source link

GPU内存不足，可否改为8位 #7

Open jia95812 opened 3 months ago

jia95812 commented 3 months ago

报错torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 512.00 MiB. GPU 可否改为8bit，如何修改？ 或者是否有其它方法？

FreeStyleFreeLunch commented 3 months ago

You can adjust the batch size smaller. We have not tried setting it to 8bit, you can try it to see if it works.