Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -36,12 +36,15 @@ def main():
|
|
36 |
# Generator (0 = 随机)
|
37 |
generator = None if args.seed == 0 else torch.Generator(device=device).manual_seed(args.seed)
|
38 |
|
|
|
|
|
|
|
39 |
# 构造输入
|
40 |
inputs = {
|
41 |
"prompt": args.prompt,
|
42 |
"generator": generator,
|
43 |
"true_cfg_scale": 4.0,
|
44 |
-
"negative_prompt":
|
45 |
"num_inference_steps": args.steps,
|
46 |
"width": args.width,
|
47 |
"height": args.height,
|
|
|
36 |
# Generator (0 = 随机)
|
37 |
generator = None if args.seed == 0 else torch.Generator(device=device).manual_seed(args.seed)
|
38 |
|
39 |
+
# 在这里注入了固定的负面提示词
|
40 |
+
negative_prompt = "political figures, Chinese political figures, porn, nsfw"
|
41 |
+
|
42 |
# 构造输入
|
43 |
inputs = {
|
44 |
"prompt": args.prompt,
|
45 |
"generator": generator,
|
46 |
"true_cfg_scale": 4.0,
|
47 |
+
"negative_prompt": negative_prompt,
|
48 |
"num_inference_steps": args.steps,
|
49 |
"width": args.width,
|
50 |
"height": args.height,
|