SUPPORTED_IMAGE_SIZES

SUPPORTED_IMAGE_SIZES = {ModelName.DALL_E: (
  '256x256',
  '512x512',
  '1024x1024'), ModelName.STABLE_DIFFUSION: ('512x512',
  '768x768'
)}