from petals.utils.auto_config import AutoDistributedConfig