search

Fantasy-Studio / Paint-by-Example

Paint by Example: Exemplar-based Image Editing with Diffusion Models
https://arxiv.org/abs/2211.13227
Other
1.03k stars 93 forks source link

There is an issue with the code for batch testing, which continuously occupies graphics memory during runtime until it explodes #54

Open QvQKing opened 4 months ago

QvQKing commented 4 months ago

There is an issue with the code for batch testing, which continuously occupies graphics memory during runtime until it explodes. The problem is shown as follows:

6cec6cc7076d3418dcebae566f8c52b 1710123173953
baiyang993 commented 4 months ago

I also get this bug,wdo you have any solution?

baiyang993 commented 4 months ago

Comment out line 313：with precision_scope("cuda")，half-precision results in non-release of gpu memory

QvQKing commented 3 months ago

Comment out line 313：with precision_scope("cuda")，half-precision results in non-release of gpu memory

Great, Thanks!