Spaces:
Running
on
Zero
Running
on
Zero
zhiweili
commited on
Commit
•
3db6256
1
Parent(s):
50358bf
test p2p
Browse files- app_haircolor_img2img.py +2 -2
app_haircolor_img2img.py
CHANGED
|
@@ -10,7 +10,7 @@ from segment_utils import(
|
|
| 10 |
restore_result,
|
| 11 |
)
|
| 12 |
from diffusers import (
|
| 13 |
-
|
| 14 |
ControlNetModel,
|
| 15 |
DDIMScheduler,
|
| 16 |
DPMSolverMultistepScheduler,
|
|
@@ -49,7 +49,7 @@ controlnet = ControlNetModel.from_pretrained(
|
|
| 49 |
torch_dtype=torch.float16,
|
| 50 |
)
|
| 51 |
|
| 52 |
-
basepipeline =
|
| 53 |
BASE_MODEL,
|
| 54 |
torch_dtype=torch.float16,
|
| 55 |
use_safetensors=True,
|
|
|
|
| 10 |
restore_result,
|
| 11 |
)
|
| 12 |
from diffusers import (
|
| 13 |
+
StableDiffusionControlNetPipeline,
|
| 14 |
ControlNetModel,
|
| 15 |
DDIMScheduler,
|
| 16 |
DPMSolverMultistepScheduler,
|
|
|
|
| 49 |
torch_dtype=torch.float16,
|
| 50 |
)
|
| 51 |
|
| 52 |
+
basepipeline = StableDiffusionControlNetPipeline.from_pretrained(
|
| 53 |
BASE_MODEL,
|
| 54 |
torch_dtype=torch.float16,
|
| 55 |
use_safetensors=True,
|