behelit2

from diffusers import DiffusionPipeline
from diffusers.utils import pt_to_pil
import torch, sys, random, gc

# A function to free up memory.
def flush():
    gc.collect()
    torch.cuda.empty_cache()

# Pass image prompts from the command line in quotes "".