cocktailpeanut commited on
Commit
e32677f
·
1 Parent(s): 5ac6e45
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ else:
17
  #device = "cuda" if torch.cuda.is_available() else "cpu"
18
 
19
  #pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, revision="refs/pr/1").to(device)
20
- pipe = DiffusionPipeline.from_pretrained("cocktailpeanut/xulf-dev", torch_dtype=dtype).to(device)
21
 
22
  MAX_SEED = np.iinfo(np.int32).max
23
  MAX_IMAGE_SIZE = 2048
 
17
  #device = "cuda" if torch.cuda.is_available() else "cpu"
18
 
19
  #pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, revision="refs/pr/1").to(device)
20
+ pipe = DiffusionPipeline.from_pretrained("cocktailpeanut/xulf-s", torch_dtype=dtype).to(device)
21
 
22
  MAX_SEED = np.iinfo(np.int32).max
23
  MAX_IMAGE_SIZE = 2048