Spaces:
No application file
No application file
import torch | |
import gradio as gr | |
from diffusers import StableDiffusionPipeline | |
class CFG: | |
device = 'cpu' | |
seed = 42 | |
Generator = torch.Generator(device).manual_seed(seed) | |
image_gen_steps = 35 | |
image_gen_model_id = 'stabilityai/stable-diffusion-2' | |
image_gen_size = (400, 400) | |
image_gen_guidance_scale = 9 | |
image_gen_model = StableDiffusionPipeline.from_pretrained( | |
CFG.image_gen_model_id, torch_dtype=torch.float32, | |
revision="fp16", use_auth_token='hf_' | |
) |