Menyu commited on
Commit
6954365
1 Parent(s): ef4fdd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -14
app.py CHANGED
@@ -46,21 +46,12 @@ def infer(
46
  seed = int(randomize_seed_fn(seed, randomize_seed))
47
  generator = torch.Generator().manual_seed(seed)
48
 
49
- (
50
- prompt_embeds
51
- , prompt_neg_embeds
52
- , pooled_prompt_embeds
53
- , negative_pooled_prompt_embeds
54
- ) = get_weighted_text_embeddings_sdxl(
55
- pipe
56
- , prompt = prompt
57
- , negative_prompt = negative_prompt
58
- )
59
  image = pipe(
60
- prompt_embeds = prompt_embeds
61
- , negative_prompt_embeds = prompt_neg_embeds
62
- , pooled_prompt_embeds = pooled_prompt_embeds
63
- , negative_pooled_prompt_embeds = negative_pooled_prompt_embeds
64
  #prompt=prompt,
65
  #negative_prompt=negative_prompt,
66
  width=width,
 
46
  seed = int(randomize_seed_fn(seed, randomize_seed))
47
  generator = torch.Generator().manual_seed(seed)
48
 
49
+ (prompt_embeds, prompt_neg_embeds, pooled_prompt_embeds, negative_pooled_prompt_embeds) = get_weighted_text_embeddings_sdxl(pipe, prompt = prompt, negative_prompt = negative_prompt)
 
 
 
 
 
 
 
 
 
50
  image = pipe(
51
+ prompt_embeds = prompt_embeds,
52
+ negative_prompt_embeds = prompt_neg_embeds,
53
+ pooled_prompt_embeds = pooled_prompt_embeds,
54
+ negative_pooled_prompt_embeds = negative_pooled_prompt_embeds,
55
  #prompt=prompt,
56
  #negative_prompt=negative_prompt,
57
  width=width,