wjs0725 commited on
Commit
2bbf3ca
·
verified ·
1 Parent(s): 323d2cd

Update flux/sampling.py

Browse files
Files changed (1) hide show
  1. flux/sampling.py +3 -0
flux/sampling.py CHANGED
@@ -30,6 +30,9 @@ def prepare(t5: HFEmbedder, clip: HFEmbedder, img: Tensor, prompt: str | list[st
30
  txt = repeat(txt, "1 ... -> bs ...", bs=bs)
31
  txt_ids = torch.zeros(bs, txt.shape[1], 3)
32
 
 
 
 
33
  vec = clip(prompt)
34
  if vec.shape[0] == 1 and bs > 1:
35
  vec = repeat(vec, "1 ... -> bs ...", bs=bs)
 
30
  txt = repeat(txt, "1 ... -> bs ...", bs=bs)
31
  txt_ids = torch.zeros(bs, txt.shape[1], 3)
32
 
33
+ print("!!!!!!!!!!!!opts.source_prompt!!!!!!!!!!!!",len(opts.source_prompt))
34
+ print("!!!!!!!!!!!!opts.source_prompt!!!!!!!!!!!!",(opts.source_prompt))
35
+ print("!!!!!!!!!!clip!!!!!!!!!",device = next(clip.parameters()).device)
36
  vec = clip(prompt)
37
  if vec.shape[0] == 1 and bs > 1:
38
  vec = repeat(vec, "1 ... -> bs ...", bs=bs)