update app.py
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ lineart_pipe = StyleContentStableDiffusionControlNetPipeline.from_pretrained(bas
|
|
| 58 |
lineart_styleshot = StyleShot(device, lineart_pipe, lineart_ip_ckpt, lineart_style_aware_encoder_path, lineart_transformer_block_path)
|
| 59 |
|
| 60 |
|
| 61 |
-
@spaces.GPU(duration=
|
| 62 |
def process(style_image, content_image, prompt, num_samples, image_resolution, condition_scale, style_scale,ddim_steps, guidance_scale, seed, a_prompt, n_prompt, btn1, Contour_Threshold=200):
|
| 63 |
weight_dtype = torch.float32
|
| 64 |
|
|
|
|
| 58 |
lineart_styleshot = StyleShot(device, lineart_pipe, lineart_ip_ckpt, lineart_style_aware_encoder_path, lineart_transformer_block_path)
|
| 59 |
|
| 60 |
|
| 61 |
+
@spaces.GPU(duration=150)
|
| 62 |
def process(style_image, content_image, prompt, num_samples, image_resolution, condition_scale, style_scale,ddim_steps, guidance_scale, seed, a_prompt, n_prompt, btn1, Contour_Threshold=200):
|
| 63 |
weight_dtype = torch.float32
|
| 64 |
|