Geek7 commited on
Commit
572328c
·
verified ·
1 Parent(s): 6f66f8c

Update myapp.py

Browse files
Files changed (1) hide show
  1. myapp.py +1 -1
myapp.py CHANGED
@@ -15,7 +15,7 @@ CORS(myapp) # Enable CORS if needed
15
  device = "cpu"
16
  dtype = torch.float32
17
 
18
- repo = "stabilityai/stable-diffusion-3-medium-diffusers"
19
  pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=dtype).to(device)
20
 
21
  MAX_SEED = np.iinfo(np.int32).max
 
15
  device = "cpu"
16
  dtype = torch.float32
17
 
18
+ repo = "prompthero/openjourney-v4"
19
  pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=dtype).to(device)
20
 
21
  MAX_SEED = np.iinfo(np.int32).max