Skip to content

GitLab

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

Update GPT2.cpp

parent a2b41d78
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
Src/GPT2.cpp
View file @ 4c121fb5
...@@ -46,7 +46,7 @@ ErrorCode GPT2::Initialize() ...@@ -46,7 +46,7 @@ ErrorCode GPT2::Initialize()
static bool CompareM(Predictions a, Predictions b) 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) long unsigned int GPT2::Inference(const std::vector<long unsigned int> &input_id)
...@@ -117,4 +117,4 @@ ErrorCode GPT2::Preprocessing(cuBERT::FullTokenizer tokenizer, ...@@ -117,4 +117,4 @@ ErrorCode GPT2::Preprocessing(cuBERT::FullTokenizer tokenizer,
} }
return SUCCESS; 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