cocktailpeanut commited on
Commit
abc0584
·
1 Parent(s): 9c06083
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -24,7 +24,8 @@ MAX_SEED = np.iinfo(np.int32).max
24
 
25
 
26
  # Initialize the pipeline and download the sd3 medium model
27
- pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
 
28
  pipe.to(device)
29
 
30
  # superprompt-v1
 
24
 
25
 
26
  # Initialize the pipeline and download the sd3 medium model
27
+ #pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
28
+ pipe = StableDiffusion3Pipeline.from_pretrained("cocktailpeanut/diffusers3ds", torch_dtype=torch.float16)
29
  pipe.to(device)
30
 
31
  # superprompt-v1