cocktailpeanut commited on
Commit
684327d
1 Parent(s): d4a9ee3
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -443,8 +443,7 @@ 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
- 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
- #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),
 
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),