ysharma HF staff commited on
Commit
621ea3a
·
1 Parent(s): 6d9201f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import pipeline_calls
6
 
7
 
8
  # init models
9
- model_ckpt = "stability/stable-diffusion-2-base"
10
  scheduler = DDIMScheduler.from_pretrained(model_ckpt, subfolder="scheduler")
11
  pipeline = StableDiffusionPanoramaPipeline.from_pretrained(
12
  model_ckpt, scheduler=scheduler, torch_dtype=torch.float16
 
6
 
7
 
8
  # init models
9
+ model_ckpt = "stabilityai/stable-diffusion-2-base"
10
  scheduler = DDIMScheduler.from_pretrained(model_ckpt, subfolder="scheduler")
11
  pipeline = StableDiffusionPanoramaPipeline.from_pretrained(
12
  model_ckpt, scheduler=scheduler, torch_dtype=torch.float16