update ddpm

README.md

@@ -4,7 +4,6 @@ emoji: 🖼️
 colorFrom: gray
 colorTo: purple
 sdk: docker
-app_file: app.py
 pinned: false
 license: mit
 ---

main.py

@@ -11,12 +11,12 @@ app = FastAPI()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model_name = "runwayml/stable-diffusion-v1-5"
 
-pipe = StableDiffusionPipeline.from_pretrained(model_name, use_safetensors=True)
-pipe = pipe.to(device)
-
 tokenizer = AutoTokenizer(model_name, subfolder="tokenizer")
 models = model_loader.from_pretrained(model_name, device)
 
+pipe = StableDiffusionPipeline.from_pretrained(model_name, use_safetensors=True)
+pipe = pipe.to(device)
+
 def generate_image(prompt, negative_prompt, seed, guidance_scale, num_inference_steps, model):
 
     if model == "from-scratch":

sd/pipeline.py

@@ -1,7 +1,7 @@
 import torch
 import numpy as np
 from tqdm import tqdm
-from ddpm import DDPMSampler
+from sd.ddpm import DDPMSampler
 
 WIDTH = 512
 HEIGHT = 512