Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,6 +20,7 @@ pipe = pipe.to(device)
|
|
| 20 |
MAX_SEED = np.iinfo(np.int32).max
|
| 21 |
MAX_IMAGE_SIZE = 300
|
| 22 |
|
|
|
|
| 23 |
|
| 24 |
# @spaces.GPU #[uncomment to use ZeroGPU]
|
| 25 |
def infer(
|
|
@@ -39,6 +40,7 @@ def infer(
|
|
| 39 |
generator = torch.Generator().manual_seed(seed)
|
| 40 |
|
| 41 |
image = pipe(
|
|
|
|
| 42 |
prompt=prompt,
|
| 43 |
negative_prompt=negative_prompt,
|
| 44 |
guidance_scale=guidance_scale,
|
|
|
|
| 20 |
MAX_SEED = np.iinfo(np.int32).max
|
| 21 |
MAX_IMAGE_SIZE = 300
|
| 22 |
|
| 23 |
+
# ပုံဖန်တီးတဲ့အခါ safety_checker ကို
|
| 24 |
|
| 25 |
# @spaces.GPU #[uncomment to use ZeroGPU]
|
| 26 |
def infer(
|
|
|
|
| 40 |
generator = torch.Generator().manual_seed(seed)
|
| 41 |
|
| 42 |
image = pipe(
|
| 43 |
+
safety_checker=None
|
| 44 |
prompt=prompt,
|
| 45 |
negative_prompt=negative_prompt,
|
| 46 |
guidance_scale=guidance_scale,
|