Spaces:
Running
Running
zhiweili
commited on
Commit
·
4a4baa6
1
Parent(s):
f0686f5
change base model
Browse files
app_haircolor_inpaint_15.py
CHANGED
|
@@ -59,8 +59,7 @@ controlnet = [
|
|
| 59 |
basepipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
|
| 60 |
BASE_MODEL,
|
| 61 |
torch_dtype=torch.float16,
|
| 62 |
-
|
| 63 |
-
revision="fp16",
|
| 64 |
controlnet=controlnet,
|
| 65 |
)
|
| 66 |
# basepipeline.scheduler = DDIMScheduler.from_config(basepipeline.scheduler.config)
|
|
|
|
| 59 |
basepipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
|
| 60 |
BASE_MODEL,
|
| 61 |
torch_dtype=torch.float16,
|
| 62 |
+
use_safetensors=True,
|
|
|
|
| 63 |
controlnet=controlnet,
|
| 64 |
)
|
| 65 |
# basepipeline.scheduler = DDIMScheduler.from_config(basepipeline.scheduler.config)
|