lixiang46
commited on
Commit
•
2502de8
1
Parent(s):
5801905
update requirements
Browse files
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 =
|
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
|