multimodalart HF staff commited on
Commit
af56f98
1 Parent(s): 3085850

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ def sample(zs, wts, prompt_tar="", cfg_scale_tar=15, skip=36, eta = 1):
57
  return img
58
 
59
  # load pipelines
60
- sd_model_id = "runwayml/stable-diffusion-v1-5"
61
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
62
  sd_pipe = StableDiffusionPipeline.from_pretrained(sd_model_id).to(device)
63
  sd_pipe.scheduler = DDIMScheduler.from_config(sd_model_id, subfolder = "scheduler")
 
57
  return img
58
 
59
  # load pipelines
60
+ sd_model_id = "stabilityai/stable-diffusion-2-base"
61
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
62
  sd_pipe = StableDiffusionPipeline.from_pretrained(sd_model_id).to(device)
63
  sd_pipe.scheduler = DDIMScheduler.from_config(sd_model_id, subfolder = "scheduler")