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