Spaces:
Running
on
T4
Running
on
T4
Commit
·
d7ce33c
1
Parent(s):
a1be0d0
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ def generate(prompt, samp_steps, batch_size, seed, progress=gr.Progress(track_tq
|
|
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,
|
33 |
if seed < 0:
|
34 |
seed = random.randint(1,999999)
|
35 |
images = pipe(
|
@@ -46,7 +46,6 @@ def set_base_model():
|
|
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")
|
|
|
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, 80085
|
33 |
if seed < 0:
|
34 |
seed = random.randint(1,999999)
|
35 |
images = pipe(
|
|
|
46 |
"stabilityai/sdxl-turbo",
|
47 |
torch_dtype = torch.float16,
|
48 |
variant = "fp16",
|
|
|
49 |
#use_auth_token=hf_token
|
50 |
)
|
51 |
pipe.to("cuda")
|