zhiweili
commited on
Commit
·
8d7237d
1
Parent(s):
6baa93c
adjust default value
Browse files- app_base.py +2 -2
- inversion_run_base.py +0 -9
app_base.py
CHANGED
@@ -81,14 +81,14 @@ def create_demo() -> gr.Blocks:
|
|
81 |
start_step = gr.Slider(minimum=1, maximum=100, value=15, step=1, label="Start Step")
|
82 |
with gr.Accordion("Advanced Options", open=False):
|
83 |
guidance_scale = gr.Slider(minimum=0, maximum=20, value=0, step=0.5, label="Guidance Scale")
|
84 |
-
generate_size = gr.Number(label="Generate Size", value=
|
85 |
mask_expansion = gr.Number(label="Mask Expansion", value=50, visible=True)
|
86 |
mask_dilation = gr.Slider(minimum=0, maximum=10, value=2, step=1, label="Mask Dilation")
|
87 |
enhance_face = gr.Checkbox(label="Enhance Face", value=False)
|
88 |
adapter_weights = gr.Slider(minimum=0, maximum=1, value=0.5, step=0.1, label="Adapter Weights", visible=False)
|
89 |
with gr.Column():
|
90 |
seed = gr.Number(label="Seed", value=8)
|
91 |
-
w1 = gr.Number(label="W1", value=2
|
92 |
g_btn = gr.Button("Edit Image")
|
93 |
|
94 |
with gr.Row():
|
|
|
81 |
start_step = gr.Slider(minimum=1, maximum=100, value=15, step=1, label="Start Step")
|
82 |
with gr.Accordion("Advanced Options", open=False):
|
83 |
guidance_scale = gr.Slider(minimum=0, maximum=20, value=0, step=0.5, label="Guidance Scale")
|
84 |
+
generate_size = gr.Number(label="Generate Size", value=1024)
|
85 |
mask_expansion = gr.Number(label="Mask Expansion", value=50, visible=True)
|
86 |
mask_dilation = gr.Slider(minimum=0, maximum=10, value=2, step=1, label="Mask Dilation")
|
87 |
enhance_face = gr.Checkbox(label="Enhance Face", value=False)
|
88 |
adapter_weights = gr.Slider(minimum=0, maximum=1, value=0.5, step=0.1, label="Adapter Weights", visible=False)
|
89 |
with gr.Column():
|
90 |
seed = gr.Number(label="Seed", value=8)
|
91 |
+
w1 = gr.Number(label="W1", value=2)
|
92 |
g_btn = gr.Button("Edit Image")
|
93 |
|
94 |
with gr.Row():
|
inversion_run_base.py
CHANGED
@@ -1,5 +1,4 @@
|
|
1 |
import torch
|
2 |
-
import os
|
3 |
|
4 |
from diffusers import (
|
5 |
DDPMScheduler,
|
@@ -12,14 +11,6 @@ from config import get_config, get_num_steps_actual
|
|
12 |
from functools import partial
|
13 |
from compel import Compel, ReturnedEmbeddingsType
|
14 |
|
15 |
-
os.system("pip freeze")
|
16 |
-
if not os.path.exists('GFPGANv1.4.pth'):
|
17 |
-
os.system("wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth -P .")
|
18 |
-
if not os.path.exists('realesr-general-x4v3.pth'):
|
19 |
-
os.system("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth -P .")
|
20 |
-
|
21 |
-
os.makedirs('output', exist_ok=True)
|
22 |
-
|
23 |
class Object(object):
|
24 |
pass
|
25 |
|
|
|
1 |
import torch
|
|
|
2 |
|
3 |
from diffusers import (
|
4 |
DDPMScheduler,
|
|
|
11 |
from functools import partial
|
12 |
from compel import Compel, ReturnedEmbeddingsType
|
13 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
14 |
class Object(object):
|
15 |
pass
|
16 |
|