Spaces:
Running
on
Zero
Running
on
Zero
zhiweili
commited on
Commit
·
7bd62db
1
Parent(s):
4a4baa6
change base model
Browse files
app_haircolor_inpaint_15.py
CHANGED
|
@@ -59,7 +59,7 @@ controlnet = [
|
|
| 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)
|
|
|
|
| 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)
|