aifeifei798 commited on
Commit
6bd743d
·
verified ·
1 Parent(s): 33c8594

Update feifeilib/feifeimodload.py

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