aifeifei798 commited on
Commit
ee8ea8d
·
verified ·
1 Parent(s): e7a2234

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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