Spaces:
Running
on
Zero
Running
on
Zero
gen5
Browse files
app.py
CHANGED
|
@@ -45,17 +45,18 @@ def infer(
|
|
| 45 |
return None
|
| 46 |
IMG_SIZE = 512
|
| 47 |
image = pipeline.generate_image(
|
| 48 |
-
|
|
|
|
| 49 |
num_images_per_caption=1,
|
| 50 |
positive_prompt=positive_prompt,
|
| 51 |
negative_prompt=negative_prompt,
|
| 52 |
-
hw=(IMG_SIZE, IMG_SIZE),
|
| 53 |
cfg=text_cfg,
|
| 54 |
cfg_img=img_cfg,
|
| 55 |
cfg_schedule="constant",
|
| 56 |
-
use_norm=
|
| 57 |
num_sampling_steps=num_inference_steps,
|
| 58 |
-
|
|
|
|
| 59 |
progress=True,
|
| 60 |
)
|
| 61 |
|
|
|
|
| 45 |
return None
|
| 46 |
IMG_SIZE = 512
|
| 47 |
image = pipeline.generate_image(
|
| 48 |
+
editing_caption,
|
| 49 |
+
hw=(IMG_SIZE, IMG_SIZE),
|
| 50 |
num_images_per_caption=1,
|
| 51 |
positive_prompt=positive_prompt,
|
| 52 |
negative_prompt=negative_prompt,
|
|
|
|
| 53 |
cfg=text_cfg,
|
| 54 |
cfg_img=img_cfg,
|
| 55 |
cfg_schedule="constant",
|
| 56 |
+
use_norm=False,
|
| 57 |
num_sampling_steps=num_inference_steps,
|
| 58 |
+
timesteps_shift=1.0,
|
| 59 |
+
seed=3407,
|
| 60 |
progress=True,
|
| 61 |
)
|
| 62 |
|