lixiang46 commited on
Commit
2502de8
1 Parent(s): 5801905

update requirements

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -80,12 +80,13 @@ def infer(prompt,
80
  negative_prompt = "nsfw,脸部阴影,低分辨率,jpeg伪影、模糊、糟糕,黑脸,霓虹灯",
81
  seed = 66,
82
  randomize_seed = False,
83
- guidance_scale = 6.0,
84
  num_inference_steps = 50
85
  ):
86
  if randomize_seed:
87
  seed = random.randint(0, MAX_SEED)
88
  generator = torch.Generator().manual_seed(seed)
 
89
  pipe = pipe.to(device)
90
  pipe.load_ip_adapter_faceid_plus(f'{ckpt_dir_faceid}/ipa-faceid-plus.bin', device = device)
91
  scale = 0.8
 
80
  negative_prompt = "nsfw,脸部阴影,低分辨率,jpeg伪影、模糊、糟糕,黑脸,霓虹灯",
81
  seed = 66,
82
  randomize_seed = False,
83
+ guidance_scale = 5.0,
84
  num_inference_steps = 50
85
  ):
86
  if randomize_seed:
87
  seed = random.randint(0, MAX_SEED)
88
  generator = torch.Generator().manual_seed(seed)
89
+ global pipe
90
  pipe = pipe.to(device)
91
  pipe.load_ip_adapter_faceid_plus(f'{ckpt_dir_faceid}/ipa-faceid-plus.bin', device = device)
92
  scale = 0.8