Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -39,13 +39,13 @@ for p in (MODELS, LORA, UPSCALE): p.mkdir(parents=True, exist_ok=True)
|
|
39 |
# 3. モデル ID / ファイル
|
40 |
##############################################################################
|
41 |
# --- BRA v7 (公開) ---
|
42 |
-
BRA_REPO = "AIGaming/beautiful_realistic_asians"
|
43 |
BRA_FILE = "beautifulRealistic_v7.safetensors"
|
44 |
BRA_REV = "801a9b1999dd7018e58a1e2b432fdccd3d1d723d" # 固定 revision
|
45 |
|
46 |
# --- IP-Adapter 本体 & LoRA ---
|
47 |
-
### 修正 ###
|
48 |
-
IP_REPO,
|
49 |
LORA_REPO,IP_LORA = "h94/IP-Adapter-FaceID", "ip-adapter-faceid-plusv2_sd15_lora.safetensors"
|
50 |
|
51 |
# --- ControlNet (MediaPipe Face) ---
|
@@ -94,10 +94,10 @@ def init():
|
|
94 |
pipe_.scheduler = DPMSolverMultistepScheduler.from_config(pipe_.scheduler.config)
|
95 |
|
96 |
# 6-4 IP-Adapter
|
97 |
-
### 修正 ### dl
|
98 |
-
|
99 |
ip_lora = dl(LORA_REPO, IP_LORA)
|
100 |
-
pipe_.load_ip_adapter(
|
101 |
AttnProcsLayers(pipe_.unet.attn_processors).load_lora_weights(
|
102 |
ip_lora, adapter_name="ip_faceid", safe_load=True
|
103 |
)
|
@@ -137,7 +137,6 @@ def generate(face: Image.Image, subj: str, add: str, neg: str,
|
|
137 |
num_inference_steps=steps, guidance_scale=cfg,
|
138 |
image=face, width=w, height=h).images[0]
|
139 |
if up:
|
140 |
-
# upsampler.scale = 4 if upf==4 else 8 <- This is also a bug, radio button gives string value
|
141 |
upsampler.scale = int(upf)
|
142 |
img, _ = upsampler.enhance(np.array(img)); img = Image.fromarray(img)
|
143 |
return img
|
@@ -155,8 +154,7 @@ with gr.Blocks(title="BRA v7 × InstantID (ZeroGPU)") as demo:
|
|
155 |
with gr.Row():
|
156 |
st = gr.Slider(10,50,30,1,"Steps"); W = gr.Slider(512,1024,768,64,"W"); H = gr.Slider(512,1024,768,64,"H")
|
157 |
with gr.Row():
|
158 |
-
|
159 |
-
up = gr.Checkbox(label="Real-ESRGAN"); upf = gr.Radio([4,8], value=4, label="アップスケール", info="Upscale factor")
|
160 |
btn = gr.Button("Generate"); out = gr.Image(type="pil", label="Result")
|
161 |
btn.click(generate, [f,s,ap,ng,cf,ip,st,W,H,up,upf], out, show_progress=True)
|
162 |
|
|
|
39 |
# 3. モデル ID / ファイル
|
40 |
##############################################################################
|
41 |
# --- BRA v7 (公開) ---
|
42 |
+
BRA_REPO = "AIGaming/beautiful_realistic_asians" # :contentReference[oaicite:0]
|
43 |
BRA_FILE = "beautifulRealistic_v7.safetensors"
|
44 |
BRA_REV = "801a9b1999dd7018e58a1e2b432fdccd3d1d723d" # 固定 revision
|
45 |
|
46 |
# --- IP-Adapter 本体 & LoRA ---
|
47 |
+
### 修正 ### 重みファイルへのフルパスを指定し、サブフォルダ定義を削除
|
48 |
+
IP_REPO, IP_BIN = "h94/IP-Adapter", "models/ip-adapter-plus-face_sd15.bin"
|
49 |
LORA_REPO,IP_LORA = "h94/IP-Adapter-FaceID", "ip-adapter-faceid-plusv2_sd15_lora.safetensors"
|
50 |
|
51 |
# --- ControlNet (MediaPipe Face) ---
|
|
|
94 |
pipe_.scheduler = DPMSolverMultistepScheduler.from_config(pipe_.scheduler.config)
|
95 |
|
96 |
# 6-4 IP-Adapter
|
97 |
+
### 修正 ### load_ip_adapter がダウンロードを処理するため、個別のdlは不要。
|
98 |
+
### また、呼び出し方を修正し、リポジトリIDと重みのフルパスを渡す。
|
99 |
ip_lora = dl(LORA_REPO, IP_LORA)
|
100 |
+
pipe_.load_ip_adapter(IP_REPO, weight_name=IP_BIN, cache_dir=str(MODELS))
|
101 |
AttnProcsLayers(pipe_.unet.attn_processors).load_lora_weights(
|
102 |
ip_lora, adapter_name="ip_faceid", safe_load=True
|
103 |
)
|
|
|
137 |
num_inference_steps=steps, guidance_scale=cfg,
|
138 |
image=face, width=w, height=h).images[0]
|
139 |
if up:
|
|
|
140 |
upsampler.scale = int(upf)
|
141 |
img, _ = upsampler.enhance(np.array(img)); img = Image.fromarray(img)
|
142 |
return img
|
|
|
154 |
with gr.Row():
|
155 |
st = gr.Slider(10,50,30,1,"Steps"); W = gr.Slider(512,1024,768,64,"W"); H = gr.Slider(512,1024,768,64,"H")
|
156 |
with gr.Row():
|
157 |
+
up = gr.Checkbox(label="Real-ESRGAN"); upf = gr.Radio([4,8], value=4, label="アップスケール")
|
|
|
158 |
btn = gr.Button("Generate"); out = gr.Image(type="pil", label="Result")
|
159 |
btn.click(generate, [f,s,ap,ng,cf,ip,st,W,H,up,upf], out, show_progress=True)
|
160 |
|