dangthr commited on
Commit
97403e9
·
verified ·
1 Parent(s): d467073

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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,