Spaces:

wl-zhao
/

unipc_sdm

Runtime error

zwl commited on Feb 17, 2023

Commit

9103864

1 Parent(s): 184c9f2

fix bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,8 +32,8 @@ if torch.cuda.is_available():
   for model in models:
     try:
         unet = UNet2DConditionModel.from_pretrained(model.path, subfolder="unet", torch_dtype=torch.float16, use_auth_token=auth_token)
-        model.pipe_t2i = StableDiffusionPipeline.from_pretrained(model.path, unet=unet, vae=vae, torch_dtype=torch.float16, scheduler=scheduler, use_auth_token=auth_token)
-        model.pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(model.path, unet=unet, vae=vae, torch_dtype=torch.float16, scheduler=scheduler, use_auth_token=auth_token)
     except:
         models.remove(model)
   pipe = models[0].pipe_t2i

   for model in models:
     try:
         unet = UNet2DConditionModel.from_pretrained(model.path, subfolder="unet", torch_dtype=torch.float16, use_auth_token=auth_token)
+        model.pipe_t2i = StableDiffusionPipeline.from_pretrained(model.path, unet=unet, vae=vae, torch_dtype=torch.float16, use_auth_token=auth_token)
+        model.pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(model.path, unet=unet, vae=vae, torch_dtype=torch.float16, use_auth_token=auth_token)
     except:
         models.remove(model)
   pipe = models[0].pipe_t2i