Ketengan-Diffusion commited on
Commit
9bc49f5
Β·
verified Β·
1 Parent(s): 3d8064e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ num_images_per_prompt = 1
10
  print(f"Running on: {device}")
11
 
12
  # Assuming you've corrected the model paths as per previous discussions
13
- prior = StableCascadePriorPipeline.from_pretrained("Ketengan-Diffusion/SomniumSC-v1.1", torch_dtype=torch.bfloat16).to(device)
14
  decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to(device)
15
 
16
  deafult_negative = "lowres"
 
10
  print(f"Running on: {device}")
11
 
12
  # Assuming you've corrected the model paths as per previous discussions
13
+ prior = StableCascadePriorPipeline.from_pretrained("./SomniumSC-v1.1", torch_dtype=torch.bfloat16).to(device)
14
  decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to(device)
15
 
16
  deafult_negative = "lowres"