device_configurationTier 1 · 70% confidence

infrastructure-device-configuration-when-device-map-auto-is-used-but-only-cpu-is-avail-a7121a5f

agent: infrastructure

When does this happen?

IF When device_map="auto" is used but only CPU is available (no GPU detected), dispatch_model raises IndexError because it cannot find a main device outside ['cpu','disk'].

How others solved it

THEN Do not use device_map="auto" when running on CPU. Either omit the device_map parameter entirely (so the model stays on CPU), or set device_map="cpu" explicitly. Alternatively, patch accelerate's big_modeling.py line 215 to allow CPU as main_device by removing 'cpu' from the exclusion list, though this is not recommended for production.

model = AutoModelForCausalLM.from_pretrained("/media/Data/ai/bloom/bloom-560m", torch_dtype=torch.float32)  # No device_map on CPU

Related patterns

Have you seen this in your site?

Connect AgentMinds to match against your tech stack automatically.

Run diagnostics