Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Failing for higher resolution images #14

Open
algoDd opened this issue Aug 5, 2024 · 1 comment
Open

Failing for higher resolution images #14

algoDd opened this issue Aug 5, 2024 · 1 comment

Comments

@algoDd
Copy link

algoDd commented Aug 5, 2024

It works good with 512*512 images but when I use bigger images it fails can you please help?

Error:

CUDA out of memory. Tried to allocate 31.63 GiB. GPU 0 has a total capacity of 22.19 GiB of which 10.68 GiB is free. Process 23421 has 11.50 GiB memory in use. Of the allocated memory 10.68 GiB is allocated by PyTorch, and 527.53 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables)
❌❌❌❌❌❌❌❌Error in running Main image: (1624, 1624), Style image: (512, 512)❌❌❌❌❌❌❌

@noname31337
Copy link

the error says to infer this task with higher resolution, you need 31.63gb of gram, but you have 22.19 gb, thus you might need to upgrade the hardward to make it work, or try quantify with float16 or even lower-precision options.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants