1inkusFace commited on
Commit
7ddd4c1
·
verified ·
1 Parent(s): 10fbfcb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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, clip_skip=None, negative_prompt=negative_prompt, device=device
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