Commit 4c121fb5 authored by yangql's avatar yangql
Browse files

Update GPT2.cpp

parent a2b41d78
......@@ -46,7 +46,7 @@ ErrorCode GPT2::Initialize()
static bool CompareM(Predictions a, Predictions b)
{
return a.predictionvalue > b.predictionvalue;
return a.predictionvalue > b.predictionvalue;
}
long unsigned int GPT2::Inference(const std::vector<long unsigned int> &input_id)
......@@ -117,4 +117,4 @@ ErrorCode GPT2::Preprocessing(cuBERT::FullTokenizer tokenizer,
}
return SUCCESS;
}
}
\ No newline at end of file
}
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