Skip to content

Commit

Permalink
remove print statement
Browse files Browse the repository at this point in the history
  • Loading branch information
JuliaGrosse committed Nov 15, 2024
1 parent 14cb82a commit 62c2d76
Showing 2 changed files with 2 additions and 1 deletion.
2 changes: 2 additions & 0 deletions examples/generate.py
Original file line number Diff line number Diff line change
@@ -72,6 +72,8 @@
max_tokens=40,
max_beam_size=5,
epsilon=0.1,
stop_at_eos=True,
use_full_budget=True,
prior_kind="dirichlet",
prior_dirichlet_alpha=0.0001,
prior_dir="./ults_priors",
1 change: 0 additions & 1 deletion ults/ults.py
Original file line number Diff line number Diff line change
@@ -410,7 +410,6 @@ def search(self) -> tuple[torch.Tensor, float, int]:
# Add the children to the tree
for i in range(self.buffer_size):
child_obs = children_observations[i]
print(child_obs)
child_name = new_node_name + "*" + str(i)
child_tokens = children_tokens[i][None, :]
penalty = (

0 comments on commit 62c2d76

Please sign in to comment.