manbeast3b
/

supatest3

Model card Files Files and versions Community

manbeast3b commited on 10 days ago

Commit

3552027

•

1 Parent(s): 7345a56

Update src/utils.py

Files changed (1) hide show

src/utils.py +0 -1

src/utils.py CHANGED Viewed

@@ -175,7 +175,6 @@ def warpped_feature(sample, step):
     step: timestep span
     """
     bs, dim, h, w = sample.shape
-    print(sample.shape)
     uncond_fea, cond_fea = sample.chunk(2)
     uncond_fea = uncond_fea.repeat(step,1,1,1) # (step * bs//2) * dim * h *w
     cond_fea = cond_fea.repeat(step,1,1,1) # (step * bs//2) * dim * h *w

     step: timestep span
     """
     bs, dim, h, w = sample.shape
     uncond_fea, cond_fea = sample.chunk(2)
     uncond_fea = uncond_fea.repeat(step,1,1,1) # (step * bs//2) * dim * h *w
     cond_fea = cond_fea.repeat(step,1,1,1) # (step * bs//2) * dim * h *w