Skip to content

GitLab

Commit 593b47b4 authored by rprenger's avatar rprenger
Browse files

Addressing comments

parent 2c500111
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
megatron/text_generation_utils.py
View file @ 593b47b4
...@@ -222,8 +222,6 @@ def generate_samples_eval(model, context, max_gen_length, eos_token_id): ...@@ -222,8 +222,6 @@ def generate_samples_eval(model, context, max_gen_length, eos_token_id):
resp_sentences = generate(model, [context], max_gen_length) resp_sentences = generate(model, [context], max_gen_length)
if resp_sentences: if resp_sentences:
return resp_sentences[0][raw_text_len:] return resp_sentences[0][raw_text_len:]
else:
return [None] # This is horrible
def switch(val1, val2, boolean): def switch(val1, val2, boolean):
boolean = boolean.type_as(val1) boolean = boolean.type_as(val1)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment