Spaces:
Sleeping
Sleeping
File size: 506 Bytes
1975737 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import torch
DEVICE = "mps"
# DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
HEIGHT = 512
WIDTH = 512
GUIDANCE_SCALE = 8
LOSS_SCALE = 200
NUM_INFERENCE_STEPS = 50
BATCH_SIZE = 1
SEEDS = [2000,2000,500,600,100]
STABLE_DIFFUSION_MODEL = "CompVis/stable-diffusion-v1-4"
STABLE_DIFUSION_CONCEPTS = ['<meeg>', '<midjourney-style>', '<moebius>', ' <Marc_Allante>', '<wlop-style>']
#LMS DSCRETE SCHEDULER
BETA_START = 0.00085
BETA_END = 0.012
BETA_SCHEDULE = "scaled_linear"
NUM_TRAIN_TIMESTEPS = 1000 |