Spaces:
Runtime error
Runtime error
Commit
·
684327d
1
Parent(s):
d4a9ee3
update
Browse files
app.py
CHANGED
|
@@ -443,8 +443,7 @@ use_safetensors= False
|
|
| 443 |
# pipe1.scheduler.set_timesteps(50)
|
| 444 |
###
|
| 445 |
pipe2 = PhotoMakerStableDiffusionXLPipeline.from_pretrained(
|
| 446 |
-
|
| 447 |
-
models_dict["Unstable"], use_safetensors=use_safetensors)
|
| 448 |
pipe2 = pipe2.to('cpu')
|
| 449 |
pipe2.load_photomaker_adapter(
|
| 450 |
os.path.dirname(photomaker_path),
|
|
@@ -457,8 +456,7 @@ pipe2.enable_freeu(s1=0.6, s2=0.4, b1=1.1, b2=1.2)
|
|
| 457 |
pipe2.fuse_lora()
|
| 458 |
|
| 459 |
pipe4 = PhotoMakerStableDiffusionXLPipeline.from_pretrained(
|
| 460 |
-
|
| 461 |
-
models_dict["RealVision"], use_safetensors=True)
|
| 462 |
pipe4 = pipe4.to('cpu')
|
| 463 |
pipe4.load_photomaker_adapter(
|
| 464 |
os.path.dirname(photomaker_path),
|
|
|
|
| 443 |
# pipe1.scheduler.set_timesteps(50)
|
| 444 |
###
|
| 445 |
pipe2 = PhotoMakerStableDiffusionXLPipeline.from_pretrained(
|
| 446 |
+
models_dict["Unstable"], torch_dtype=torch.float16, use_safetensors=use_safetensors)
|
|
|
|
| 447 |
pipe2 = pipe2.to('cpu')
|
| 448 |
pipe2.load_photomaker_adapter(
|
| 449 |
os.path.dirname(photomaker_path),
|
|
|
|
| 456 |
pipe2.fuse_lora()
|
| 457 |
|
| 458 |
pipe4 = PhotoMakerStableDiffusionXLPipeline.from_pretrained(
|
| 459 |
+
models_dict["RealVision"], torch_dtype=torch.float16, use_safetensors=True)
|
|
|
|
| 460 |
pipe4 = pipe4.to('cpu')
|
| 461 |
pipe4.load_photomaker_adapter(
|
| 462 |
os.path.dirname(photomaker_path),
|