Use public pytorch image, install cosmos-guardrail with --no-deps 1c3a936 verified samwell commited on Jan 16