Hugging Face's logo

Spaces:
aifeifei798
/
DarkIdol-flux
Running on Zero

App Files
aifeifei798 commited on
Commit
36bb891
•
1 Parent(s): c225f9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -12,10 +12,15 @@ taef1 = AutoencoderTiny.from_pretrained("aifeifei798/taef1", torch_dtype=dtype).
12
  device
13
  )
14
 
 
 
 
 
15
  pipe = DiffusionPipeline.from_pretrained(
16
- "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype, vae=taef1
17
  ).to(device)
18
 
 
19
  MAX_SEED = np.iinfo(np.int32).max
20
  MAX_IMAGE_SIZE = 2048
21
 
 
12
  device
13
  )
14
 
15
+ #pipe = DiffusionPipeline.from_pretrained(
16
+ # "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype, vae=taef1
17
+ #).to(device)
18
+
19
  pipe = DiffusionPipeline.from_pretrained(
20
+ "aifeifei798/DarkIdol-flux-v1.1", torch_dtype=dtype
21
  ).to(device)
22
 
23
+
24
  MAX_SEED = np.iinfo(np.int32).max
25
  MAX_IMAGE_SIZE = 2048
26