Spaces:
Running
on
Zero
Running
on
Zero
Update utils/interface.py
Browse files- utils/interface.py +1 -1
utils/interface.py
CHANGED
|
@@ -91,7 +91,7 @@ class Inference:
|
|
| 91 |
|
| 92 |
@spaces.GPU(duration = 120)
|
| 93 |
def gen_spaces():
|
| 94 |
-
print(f"{inputs[
|
| 95 |
token_ids = self.model.generate(
|
| 96 |
input_ids=inputs["input_ids"].to(self.device),
|
| 97 |
attention_mask=inputs["attention_mask"].to(self.device),
|
|
|
|
| 91 |
|
| 92 |
@spaces.GPU(duration = 120)
|
| 93 |
def gen_spaces():
|
| 94 |
+
print(f"{inputs['input_ids'].tolist()=}")
|
| 95 |
token_ids = self.model.generate(
|
| 96 |
input_ids=inputs["input_ids"].to(self.device),
|
| 97 |
attention_mask=inputs["attention_mask"].to(self.device),
|