MODEL_GPU_TEXT: str = 'bigscience/Bloom-560m'
MODEL_CPU_TEXT: str = 'openlm-research/open_llama_3b'

MODEL_GPU_TEXT2IMAGE: str = 'stabilityai/stable-diffusion-2'
MODEL_CPU_TEXT2IMAGE: str = 'stabilityai/stable-diffusion-xl-base-1.0'
