deep-floyd / IF

Other
7.64k stars 497 forks source link

Error when running image variation section in Notebook: #47

Open Tylersuard opened 1 year ago

Tylersuard commented 1 year ago

ValueError: Some modules are dispatched on the CPU or the disk. Make sure you have enough GPU RAM to fit the quantized model. If you want to dispatch the model on the CPU or the disk while keeping these modules in 32-bit, you need to set load_in_8bit_fp32_cpu_offload=True and pass a custom device_map to from_pretrained. Check https://huggingface.co/docs/transformers/main/en/main_classes/quantization#offload-between- cpu-and-gpu for more details.

Tylersuard commented 1 year ago

Running on Colab with a GPU and high ram environment. This code caused the error:

from PIL import Image from io import BytesIO

original_image = Image.open(BytesIO(response.content)).convert("RGB") original_image = original_image.resize((768, 512)) original_image