aifeifei798 commited on
Commit
0eaa47c
·
verified ·
1 Parent(s): 6e86490

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -27
app.py CHANGED
@@ -25,10 +25,10 @@ def feifeimodload():
25
  "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
26
  ).to(device)
27
 
28
- pipe.load_lora_weights(
29
- hf_hub_download("aifeifei798/feifei-flux-lora-v1.1", "feifei-v1.1.safetensors"),
30
- adapter_name="feifei",
31
- )
32
  pipe.load_lora_weights(
33
  hf_hub_download("aifeifei798/sldr_flux_nsfw_v2-studio", "sldr_flux_nsfw_v2-studio.safetensors"),
34
  adapter_name="sldr_flux_nsfw_v2",
@@ -50,14 +50,14 @@ def feifeimodload():
50
  pipe = feifeimodload()
51
 
52
  @spaces.GPU()
53
- def infer(prompt="", styles_Radio="(None)", feife_select=False, bigboobs_select = True, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, guidancescale=3.5, num_feifei=0.35, nsfw_select=False, nsfw_slider=1, progress=gr.Progress(track_tqdm=True)):
54
 
55
  Duke86Syl_lora_name=[]
56
  adapter_weights_num=[]
57
 
58
- if feife_select:
59
- Duke86Syl_lora_name.append("feifei")
60
- adapter_weights_num.append(num_feifei)
61
 
62
  if bigboobs_select:
63
  Duke86Syl_lora_name.append("big-boobs-clothed-v2")
@@ -76,15 +76,15 @@ def infer(prompt="", styles_Radio="(None)", feife_select=False, bigboobs_select
76
  lora_scale=1.0,
77
  )
78
 
79
- if feife_select:
80
- # Define the replacement string
81
- replacement = " feifei, A beautiful, 18 yo kpop idol, large-busted Japanese slim girl, with light makeup, gazing deeply into the camera, "
82
-
83
- # Perform the replacements with re.IGNORECASE
84
- prompt = re.sub(r"girl", replacement, prompt, flags=re.IGNORECASE)
85
- prompt = re.sub(r"young woman", replacement, prompt, flags=re.IGNORECASE)
86
- prompt = re.sub(r"woman", replacement, prompt, flags=re.IGNORECASE)
87
- prompt = re.sub(r"model", replacement, prompt, flags=re.IGNORECASE)
88
 
89
  if styles_Radio:
90
  style_name = styles_Radio
@@ -94,16 +94,7 @@ def infer(prompt="", styles_Radio="(None)", feife_select=False, bigboobs_select
94
  if randomize_seed:
95
  seed = random.randint(0, MAX_SEED)
96
  generator = torch.Generator().manual_seed(seed)
97
-
98
- pipe.set_adapters(
99
- ["feifei"],
100
- adapter_weights=[num_feifei],
101
- )
102
- pipe.fuse_lora(
103
- adapter_name=["feifei"],
104
- lora_scale=1.0,
105
- )
106
-
107
  image = pipe(
108
  prompt = "flux, 8k, ",
109
  prompt_2 = prompt,
 
25
  "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
26
  ).to(device)
27
 
28
+ #pipe.load_lora_weights(
29
+ # hf_hub_download("aifeifei798/feifei-flux-lora-v1.1", "feifei-v1.1.safetensors"),
30
+ # adapter_name="feifei",
31
+ #)
32
  pipe.load_lora_weights(
33
  hf_hub_download("aifeifei798/sldr_flux_nsfw_v2-studio", "sldr_flux_nsfw_v2-studio.safetensors"),
34
  adapter_name="sldr_flux_nsfw_v2",
 
50
  pipe = feifeimodload()
51
 
52
  @spaces.GPU()
53
+ def infer(prompt="", styles_Radio="(None)", feife_select = False, bigboobs_select = True, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, guidancescale=3.5, num_feifei=0.35, nsfw_select=False, nsfw_slider=1, progress=gr.Progress(track_tqdm=True)):
54
 
55
  Duke86Syl_lora_name=[]
56
  adapter_weights_num=[]
57
 
58
+ #if feife_select:
59
+ # Duke86Syl_lora_name.append("feifei")
60
+ # adapter_weights_num.append(num_feifei)
61
 
62
  if bigboobs_select:
63
  Duke86Syl_lora_name.append("big-boobs-clothed-v2")
 
76
  lora_scale=1.0,
77
  )
78
 
79
+ #if feife_select:
80
+ # # Define the replacement string
81
+ # replacement = " feifei, A beautiful, 18 yo kpop idol, large-busted Japanese slim girl, with light makeup, gazing deeply into the camera, "
82
+ #
83
+ # # Perform the replacements with re.IGNORECASE
84
+ # prompt = re.sub(r"girl", replacement, prompt, flags=re.IGNORECASE)
85
+ # prompt = re.sub(r"young woman", replacement, prompt, flags=re.IGNORECASE)
86
+ # prompt = re.sub(r"woman", replacement, prompt, flags=re.IGNORECASE)
87
+ # prompt = re.sub(r"model", replacement, prompt, flags=re.IGNORECASE)
88
 
89
  if styles_Radio:
90
  style_name = styles_Radio
 
94
  if randomize_seed:
95
  seed = random.randint(0, MAX_SEED)
96
  generator = torch.Generator().manual_seed(seed)
97
+
 
 
 
 
 
 
 
 
 
98
  image = pipe(
99
  prompt = "flux, 8k, ",
100
  prompt_2 = prompt,