Phospheneser commited on
Commit
a88ecfd
·
verified ·
1 Parent(s): f44cf4d

Update utils/interface.py

Browse files
Files changed (1) hide show
  1. 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["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),
 
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),