Skip to main content

All Questions

1 vote
0 answers
101 views

How to wipe gradients from UNet2DConditionModel

I am working with the "CompVis/ldm-text2im-large-256" building on top of the prompt-to-prompt code. model = DiffusionPipeline.from_pretrained(model_id, height=IMAGE_RES, width=IMAGE_RES).to(...
python_noob's user avatar