Spaces:

qyoo
/

Conceptrol

Running on Zero

qyoo commited on Mar 8

Commit

297dfd6

1 Parent(s): 6656488

update

Files changed (2) hide show

omini_control/__pycache__/condition.cpython-310.pyc CHANGED Viewed

Binary files a/omini_control/__pycache__/condition.cpython-310.pyc and b/omini_control/__pycache__/condition.cpython-310.pyc differ

omini_control/condition.py CHANGED Viewed

@@ -102,6 +102,8 @@ class Condition(object):
             pipe.device,
             pipe.dtype,
         )
         return cond_tokens, cond_ids
     def encode(self, pipe: FluxPipeline) -> Tuple[torch.Tensor, torch.Tensor, int]:

             pipe.device,
             pipe.dtype,
         )
+        print("Cond_tokens.shape", cond_tokens.shape)
+        print("Cond_ids.shape", cond_ids.shape)
         return cond_tokens, cond_ids
     def encode(self, pipe: FluxPipeline) -> Tuple[torch.Tensor, torch.Tensor, int]: