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,
|