Commit ·
75c3741
1
Parent(s): f5605e2
cascada
Browse files- handler.py +1 -4
handler.py
CHANGED
|
@@ -21,9 +21,6 @@ from pipeline_stable_diffusion_xl_instantid_full import StableDiffusionXLInstant
|
|
| 21 |
from controlnet_aux import OpenposeDetector
|
| 22 |
from huggingface_hub import hf_hub_download
|
| 23 |
|
| 24 |
-
|
| 25 |
-
# end_test
|
| 26 |
-
|
| 27 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 28 |
if device.type != 'cuda':
|
| 29 |
raise ValueError("Se requiere ejecutar en GPU")
|
|
@@ -151,7 +148,7 @@ class EndpointHandler():
|
|
| 151 |
|
| 152 |
|
| 153 |
identitynet_strength_ratio = 0.8
|
| 154 |
-
pose_strength = 0.
|
| 155 |
canny_strength = 0.3
|
| 156 |
# depth_strength = 0.5
|
| 157 |
self.my_controlnet_selection = ["pose", "canny"]
|
|
|
|
| 21 |
from controlnet_aux import OpenposeDetector
|
| 22 |
from huggingface_hub import hf_hub_download
|
| 23 |
|
|
|
|
|
|
|
|
|
|
| 24 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 25 |
if device.type != 'cuda':
|
| 26 |
raise ValueError("Se requiere ejecutar en GPU")
|
|
|
|
| 148 |
|
| 149 |
|
| 150 |
identitynet_strength_ratio = 0.8
|
| 151 |
+
pose_strength = 0.5
|
| 152 |
canny_strength = 0.3
|
| 153 |
# depth_strength = 0.5
|
| 154 |
self.my_controlnet_selection = ["pose", "canny"]
|