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

Update flux/sampling.py

Browse files
Files changed (1) hide show
  1. flux/sampling.py +2 -2
flux/sampling.py CHANGED
@@ -30,8 +30,8 @@ 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
- 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:
 
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(prompt))
34
+ print("!!!!!!!!!!!!opts.source_prompt!!!!!!!!!!!!",prompt)
35
  print("!!!!!!!!!!clip!!!!!!!!!",device = next(clip.parameters()).device)
36
  vec = clip(prompt)
37
  if vec.shape[0] == 1 and bs > 1: