Spaces:
Runtime error
Runtime error
cocktailpeanut
commited on
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),
|