aifeifei798 commited on
Commit
cf0664e
·
verified ·
1 Parent(s): 45c56ec

Update feifeilib/feifeimodload.py

Browse files
Files changed (1) hide show
  1. feifeilib/feifeimodload.py +1 -9
feifeilib/feifeimodload.py CHANGED
@@ -10,17 +10,9 @@ def feifeimodload():
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
 
12
  pipe = DiffusionPipeline.from_pretrained(
13
- "aifeifei798/DarkIdol-flux-v1.1", torch_dtype=dtype
14
  ).to(device)
15
 
16
- #pipe.load_lora_weights(
17
- # hf_hub_download(
18
- # "aifeifei798/sldr_flux_nsfw_v2-studio",
19
- # "sldr_flux_nsfw_v2-studio.safetensors",
20
- # ),
21
- # adapter_name="sldr_flux_nsfw_v2",
22
- #)
23
-
24
  pipe.vae.enable_slicing()
25
  pipe.vae.enable_tiling()
26
  torch.cuda.empty_cache()
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
 
12
  pipe = DiffusionPipeline.from_pretrained(
13
+ "aifeifei798/DarkIdol-flux-v1.2", torch_dtype=dtype
14
  ).to(device)
15
 
 
 
 
 
 
 
 
 
16
  pipe.vae.enable_slicing()
17
  pipe.vae.enable_tiling()
18
  torch.cuda.empty_cache()