mirror of
https://github.com/AvaLovelace1/LegoGPT.git
synced 2026-05-14 19:43:15 -05:00
remove print statement
This commit is contained in:
@@ -164,7 +164,6 @@ class LegoGPT:
|
||||
brick, rejection_reasons_brick = self.generate_brick_with_rejection_sampling(
|
||||
prompt if brick_num == 0 else None, lego=starting_lego
|
||||
)
|
||||
print(brick)
|
||||
if not brick: # EOS token was generated
|
||||
break
|
||||
rejection_reasons.update(rejection_reasons_brick)
|
||||
|
||||
Reference in New Issue
Block a user