Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,6 +61,7 @@ def init_predictor():
|
|
| 61 |
text_encoder=text_encoder,
|
| 62 |
torch_dtype=torch.bfloat16,
|
| 63 |
).to("cpu")
|
|
|
|
| 64 |
pipe.to(device)
|
| 65 |
|
| 66 |
negative_prompt = "Aerial view, aerial view, overexposed, low quality, deformation, a poor composition, bad hands, bad teeth, bad eyes, bad limbs, distortion"
|
|
@@ -71,7 +72,7 @@ def generate(segment, image, prompt, size, guidance_scale, num_inference_steps,
|
|
| 71 |
seed = int(random.randrange(4294967294))
|
| 72 |
if segment==1:
|
| 73 |
prompt_embeds, prompt_attention_mask, negative_prompt_embeds, negative_attention_mask, pooled_prompt_embeds, negative_pooled_prompt_embeds = pipe.encode_prompt(
|
| 74 |
-
prompt=prompt, do_classifier_free_guidance=True,
|
| 75 |
)
|
| 76 |
|
| 77 |
transformer_dtype = pipe.transformer.dtype
|
|
|
|
| 61 |
text_encoder=text_encoder,
|
| 62 |
torch_dtype=torch.bfloat16,
|
| 63 |
).to("cpu")
|
| 64 |
+
pipe.clip_skip=None
|
| 65 |
pipe.to(device)
|
| 66 |
|
| 67 |
negative_prompt = "Aerial view, aerial view, overexposed, low quality, deformation, a poor composition, bad hands, bad teeth, bad eyes, bad limbs, distortion"
|
|
|
|
| 72 |
seed = int(random.randrange(4294967294))
|
| 73 |
if segment==1:
|
| 74 |
prompt_embeds, prompt_attention_mask, negative_prompt_embeds, negative_attention_mask, pooled_prompt_embeds, negative_pooled_prompt_embeds = pipe.encode_prompt(
|
| 75 |
+
prompt=prompt, do_classifier_free_guidance=True, negative_prompt=negative_prompt, device=device
|
| 76 |
)
|
| 77 |
|
| 78 |
transformer_dtype = pipe.transformer.dtype
|