update app.py
Browse files
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
from types import MethodType
|
| 2 |
|
| 3 |
import os
|
|
@@ -57,6 +58,7 @@ lineart_pipe = StyleContentStableDiffusionControlNetPipeline.from_pretrained(bas
|
|
| 57 |
lineart_styleshot = StyleShot(device, lineart_pipe, lineart_ip_ckpt, lineart_style_aware_encoder_path, lineart_transformer_block_path)
|
| 58 |
|
| 59 |
|
|
|
|
| 60 |
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):
|
| 61 |
weight_dtype = torch.float32
|
| 62 |
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
from types import MethodType
|
| 3 |
|
| 4 |
import os
|
|
|
|
| 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=120)
|
| 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 |
|