Krebzonide commited on
Commit
a1be0d0
·
1 Parent(s): 2d4bfe7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -26,7 +26,7 @@ def generate(prompt, samp_steps, batch_size, seed, progress=gr.Progress(track_tq
26
  if prompt[:10] == "krebzonide":
27
  prompt = prompt[10:]
28
  else:
29
- neg_prompt = neg_prompt + ", child, nsfw, nipples, nude, underwear"
30
  for word in naughtyWords:
31
  if prompt.find(word) >= 0:
32
  return None, 58008
@@ -45,7 +45,8 @@ def set_base_model():
45
  pipe = AutoPipelineForText2Image.from_pretrained(
46
  "stabilityai/sdxl-turbo",
47
  torch_dtype = torch.float16,
48
- variant = "fp16"
 
49
  #use_auth_token=hf_token
50
  )
51
  pipe.to("cuda")
 
26
  if prompt[:10] == "krebzonide":
27
  prompt = prompt[10:]
28
  else:
29
+ neg_prompt = neg_prompt + ", child, nsfw, nipples, nude, underwear, naked"
30
  for word in naughtyWords:
31
  if prompt.find(word) >= 0:
32
  return None, 58008
 
45
  pipe = AutoPipelineForText2Image.from_pretrained(
46
  "stabilityai/sdxl-turbo",
47
  torch_dtype = torch.float16,
48
+ variant = "fp16",
49
+ height = 768
50
  #use_auth_token=hf_token
51
  )
52
  pipe.to("cuda")