cocktailpeanut commited on
Commit
d4a9ee3
1 Parent(s): 1f70aa3
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -443,7 +443,8 @@ use_safetensors= False
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,7 +457,8 @@ pipe2.enable_freeu(s1=0.6, s2=0.4, b1=1.1, b2=1.2)
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),
 
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
+ models_dict["Unstable"], use_safetensors=use_safetensors)
448
  pipe2 = pipe2.to('cpu')
449
  pipe2.load_photomaker_adapter(
450
  os.path.dirname(photomaker_path),
 
457
  pipe2.fuse_lora()
458
 
459
  pipe4 = PhotoMakerStableDiffusionXLPipeline.from_pretrained(
460
+ #models_dict["RealVision"], torch_dtype=torch.float16, use_safetensors=True)
461
+ models_dict["RealVision"], use_safetensors=True)
462
  pipe4 = pipe4.to('cpu')
463
  pipe4.load_photomaker_adapter(
464
  os.path.dirname(photomaker_path),