Spaces:
Running
Running
import random | |
import requests | |
import torch | |
from io import BytesIO | |
from PIL import Image | |
from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, StableDiffusionInpaintPipeline, StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipeline, StableDiffusionXLInpaintPipeline, DiffusionPipeline | |
import os | |
import time | |
from dotenv import load_dotenv | |
load_dotenv("config.txt") | |
xl = os.getenv("xl") | |
with torch.no_grad(): | |
pipe = StableDiffusionPipeline.from_single_file( | |
"models/checkpoint/gpu-model/base/dreamdrop-v1.safetensors", | |
use_safetensors=True, | |
cache_dir="models/checkpoint/gpu-model/base/cache_dir", | |
scheduler_type="euler-ancestral" | |
) | |
time.sleep(20) | |
pipe_inpaint = StableDiffusionInpaintPipeline.from_single_file( | |
"models/checkpoint/gpu-model/inpaint/dreamdrop-inpainting.safetensors", | |
use_safetensors=True, | |
cache_dir="models/checkpoint/gpu-model/inpaint/cache_dir", | |
scheduler_type="euler-ancestral" | |
) | |