Update README.md
Browse files
README.md
CHANGED
|
@@ -40,15 +40,6 @@ prompt = "Yoda, avatarart style person"
|
|
| 40 |
scale = 7.5
|
| 41 |
n_samples = 4
|
| 42 |
|
| 43 |
-
# Sometimes the nsfw checker is confused by the Naruto images, you can disable
|
| 44 |
-
# it at your own risk here
|
| 45 |
-
disable_safety = False
|
| 46 |
-
|
| 47 |
-
if disable_safety:
|
| 48 |
-
def null_safety(images, **kwargs):
|
| 49 |
-
return images, False
|
| 50 |
-
pipe.safety_checker = null_safety
|
| 51 |
-
|
| 52 |
with autocast("cuda"):
|
| 53 |
images = pipe(n_samples*[prompt], guidance_scale=scale).images
|
| 54 |
|
|
|
|
| 40 |
scale = 7.5
|
| 41 |
n_samples = 4
|
| 42 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
with autocast("cuda"):
|
| 44 |
images = pipe(n_samples*[prompt], guidance_scale=scale).images
|
| 45 |
|