Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -249,8 +249,8 @@ if torch.cuda.is_available():
|
|
| 249 |
# vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
| 250 |
token = os.environ.get("HF_TOKEN") # 从环境变量读取令牌
|
| 251 |
model_path = hf_hub_download(
|
| 252 |
-
repo_id="Menyu/
|
| 253 |
-
filename="
|
| 254 |
use_auth_token=token
|
| 255 |
)
|
| 256 |
pipe = StableDiffusionXLPipeline.from_single_file(
|
|
@@ -328,9 +328,9 @@ def infer(
|
|
| 328 |
"height": height,
|
| 329 |
"guidance_scale": guidance_scale,
|
| 330 |
"num_inference_steps": num_inference_steps,
|
| 331 |
-
"model": "
|
| 332 |
"use_resolution_binning": use_resolution_binning,
|
| 333 |
-
"PreUrl": "https://huggingface.co/spaces/Menyu/
|
| 334 |
}
|
| 335 |
# Add metadata to the image
|
| 336 |
image_with_metadata = add_metadata_to_image(image, metadata)
|
|
@@ -353,7 +353,7 @@ h1{text-align:center}
|
|
| 353 |
|
| 354 |
with gr.Blocks(css=css) as demo:
|
| 355 |
gr.Markdown("""# 梦羽的模型生成器
|
| 356 |
-
### 快速生成
|
| 357 |
with gr.Group():
|
| 358 |
with gr.Row():
|
| 359 |
prompt = gr.Text(
|
|
|
|
| 249 |
# vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
| 250 |
token = os.environ.get("HF_TOKEN") # 从环境变量读取令牌
|
| 251 |
model_path = hf_hub_download(
|
| 252 |
+
repo_id="Menyu/noob-vpre065s", # 模型仓库名称(非完整URL)
|
| 253 |
+
filename="miaomiaoRealskin_vPredV10.safetensors",
|
| 254 |
use_auth_token=token
|
| 255 |
)
|
| 256 |
pipe = StableDiffusionXLPipeline.from_single_file(
|
|
|
|
| 328 |
"height": height,
|
| 329 |
"guidance_scale": guidance_scale,
|
| 330 |
"num_inference_steps": num_inference_steps,
|
| 331 |
+
"model": "miaomiaoRealSkin10",
|
| 332 |
"use_resolution_binning": use_resolution_binning,
|
| 333 |
+
"PreUrl": "https://huggingface.co/spaces/Menyu/miaomiaoRealSkin"
|
| 334 |
}
|
| 335 |
# Add metadata to the image
|
| 336 |
image_with_metadata = add_metadata_to_image(image, metadata)
|
|
|
|
| 353 |
|
| 354 |
with gr.Blocks(css=css) as demo:
|
| 355 |
gr.Markdown("""# 梦羽的模型生成器
|
| 356 |
+
### 快速生成 MiaomiaoRealSkin vPred 1.0 模型的图片""")
|
| 357 |
with gr.Group():
|
| 358 |
with gr.Row():
|
| 359 |
prompt = gr.Text(
|