Fix: ensure_on_device before each inference to prevent text_encoder drift d95102e verified samwell commited on Jan 17
Explicitly move all pipeline components to CUDA, use inference_mode 850ecac verified samwell commited on Jan 17
Remove generator entirely to fix CUDA/CPU device mismatch 0c995b9 verified samwell commited on Jan 17
Fix generator - remove device spec, use official example pattern 0e4048f verified samwell commited on Jan 17
Use public pytorch image, install cosmos-guardrail with --no-deps 1c3a936 verified samwell commited on Jan 16