Spaces:
Running
on
T4
Running
on
T4
Commit
·
a1be0d0
1
Parent(s):
2d4bfe7
Update app.py
Browse files
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")
|