debug: output ids

utils/interface.py

@@ -91,6 +91,7 @@ class Inference:
 
         @spaces.GPU(duration = 120)
         def gen_spaces():
+            print(f"{inputs["input_ids"].tolist()=}")
             token_ids = self.model.generate(
                 input_ids=inputs["input_ids"].to(self.device),
                 attention_mask=inputs["attention_mask"].to(self.device),
@@ -98,6 +99,8 @@ class Inference:
                 stopping_criteria=stopping_criteria
             )
 
+            print(f"{token_ids.tolist()=}")
+
             results = self.processor.decode(
                 token_ids.to(self.device),
                 output_modalities,