Spaces:
Runtime error
Runtime error
fix steps
Browse files
app.py
CHANGED
|
@@ -83,6 +83,7 @@ def range_loss(input):
|
|
| 83 |
def inference(text, init_image, skip_timesteps, clip_guidance_scale, tv_scale, range_scale, init_scale, seed, image_prompts,timestep_respacing, cutn, im_prompt_weight):
|
| 84 |
# Model settings
|
| 85 |
skip_timesteps = min(skip_timesteps, timestep_respacing-1)
|
|
|
|
| 86 |
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
| 87 |
model_config = model_and_diffusion_defaults()
|
| 88 |
model_config.update({
|
|
@@ -247,8 +248,8 @@ with demo:
|
|
| 247 |
# with gr.Group():
|
| 248 |
with gr.Column():
|
| 249 |
init_image = gr.Image(source="upload", label='initial image (optional)')
|
| 250 |
-
init_scale = gr.Slider(minimum=0, maximum=
|
| 251 |
-
skip_timesteps = gr.Slider(minimum=0, maximum=100, step=1, value=30, label="Style strength")
|
| 252 |
# with gr.Group():
|
| 253 |
with gr.Column():
|
| 254 |
image_prompts = gr.Image(source="upload", label='image prompt (optional)')
|
|
|
|
| 83 |
def inference(text, init_image, skip_timesteps, clip_guidance_scale, tv_scale, range_scale, init_scale, seed, image_prompts,timestep_respacing, cutn, im_prompt_weight):
|
| 84 |
# Model settings
|
| 85 |
skip_timesteps = min(skip_timesteps, timestep_respacing-1)
|
| 86 |
+
skip_timesteps = int(timestep_respacing-1 - (timestep_respacing-1)*skip_timesteps/100)
|
| 87 |
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
| 88 |
model_config = model_and_diffusion_defaults()
|
| 89 |
model_config.update({
|
|
|
|
| 248 |
# with gr.Group():
|
| 249 |
with gr.Column():
|
| 250 |
init_image = gr.Image(source="upload", label='initial image (optional)')
|
| 251 |
+
init_scale = gr.Slider(minimum=0, maximum=1000, step=10, value=0, label="Look like the image above")
|
| 252 |
+
skip_timesteps = gr.Slider(minimum=0, maximum=100, step=1, value=30, label="Style strength, % (0 = initial image)")
|
| 253 |
# with gr.Group():
|
| 254 |
with gr.Column():
|
| 255 |
image_prompts = gr.Image(source="upload", label='image prompt (optional)')
|