Apppublic

bala1802/StableDiffusionModel

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face

0likes

config.py20 linesDownload Raw Back to root

1import torch2 3# DEVICE = "mps"4DEVICE = "cuda" if torch.cuda.is_available() else "cpu"5HEIGHT = 5126WIDTH = 5127GUIDANCE_SCALE = 88LOSS_SCALE = 2009NUM_INFERENCE_STEPS = 5010BATCH_SIZE = 111 12SEEDS = [2000,2000,500,600,100]13STABLE_DIFFUSION_MODEL = "CompVis/stable-diffusion-v1-4"14STABLE_DIFUSION_CONCEPTS = ['<meeg>', '<midjourney-style>', '<moebius>', ' <Marc_Allante>', '<wlop-style>']15 16#LMS DSCRETE SCHEDULER17BETA_START = 0.0008518BETA_END = 0.01219BETA_SCHEDULE = "scaled_linear"20NUM_TRAIN_TIMESTEPS = 1000