Spaces:
Running
on
Zero
Running
on
Zero
aifeifei798
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -8,9 +8,9 @@ from diffusers import DiffusionPipeline, AutoencoderTiny
|
|
8 |
dtype = torch.bfloat16
|
9 |
|
10 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
11 |
-
taef1 = AutoencoderTiny.from_pretrained("aifeifei798/taef1", torch_dtype=dtype).to(
|
12 |
-
device
|
13 |
-
)
|
14 |
|
15 |
#pipe = DiffusionPipeline.from_pretrained(
|
16 |
# "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype, vae=taef1
|
|
|
8 |
dtype = torch.bfloat16
|
9 |
|
10 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
11 |
+
#taef1 = AutoencoderTiny.from_pretrained("aifeifei798/taef1", torch_dtype=dtype).to(
|
12 |
+
# device
|
13 |
+
#)
|
14 |
|
15 |
#pipe = DiffusionPipeline.from_pretrained(
|
16 |
# "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype, vae=taef1
|