Spaces:
Running
on
A10G
RuntimeError: CUDA error: unknown error
Hello kind stranger
Thanks a lot for the code and model
I was trying to run the code here locally in a docker container, everything is working perfectly but whenever I change sample count to anything other than "1" my PC freezes for a second then it gives an error with RuntimeError: CUDA error: unknown error
.
I'm using a 3080ti and I know this might not have enough vram to run such thing but I've already ran some stable diffusion projects locally and set samples to multiple.
The following line is the only thing related I can see in the stacktrace
File "/app/lambda-diffusers/lambda_diffusers/pipelines/pipeline_stable_diffusion_im_embed.py", line 136, in __call__
noise_pred = self.unet(latent_model_input, t, encoder_hidden_states=image_embeddings)["sample"]
Thanks again for model and knowledge
I tried today with resolution 256x256 and it worked for multiple samples, but it appears that this is a memory problem, not sure why the error didn't say so though...