aifeifei798 commited on
Commit
5d2ac0b
·
verified ·
1 Parent(s): 4b5bf5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,9 +14,9 @@ def feifeimodload():
14
 
15
  dtype = torch.bfloat16
16
  device = "cuda" if torch.cuda.is_available() else "cpu"
17
- taef1 = AutoencoderTiny.from_pretrained("aifeifei798/taef1", torch_dtype=dtype).to(
18
- device
19
- )
20
 
21
  #pipe = DiffusionPipeline.from_pretrained(
22
  # "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype, vae=taef1
 
14
 
15
  dtype = torch.bfloat16
16
  device = "cuda" if torch.cuda.is_available() else "cpu"
17
+ #taef1 = AutoencoderTiny.from_pretrained("aifeifei798/taef1", torch_dtype=dtype).to(
18
+ # device
19
+ #)
20
 
21
  #pipe = DiffusionPipeline.from_pretrained(
22
  # "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype, vae=taef1