-
Notifications
You must be signed in to change notification settings - Fork 531
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
Type mismatch in gradio_app/app.py #67
Comments
Most likely, the same issue here: |
Ran into the following error when generating the image - occurs at image generation (I get the first 2 progress bars) torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 768.00 MiB. GPU 0 has a total capacty of 23.62 GiB of which 4.38 MiB is free. Including non-PyTorch memory, this process has 23.61 GiB memory in use. Of the allocated memory 22.72 GiB is allocated by PyTorch, and 423.33 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
changed line to 'c_dtype = torch.bfloat16 if device.type == "cpu" else torch.bfloat16' Image generates now without issue (much quicker as well!!!) |
same issue |
refer other issue fix this #58 have a try. |
I don't think this issue is caused by the problem I mentioned. |
Works for me. Thanks! |
This fix worked for me. Please update the README and huggingface with the correct version of diffusers. |
we fixed all errors with fp16 support : #125 |
In gradio_app/app.py, line 12
Isn't the cuda device that uses bfloat16?
The text was updated successfully, but these errors were encountered: