Skip to content

Commit fd8018e

Browse files
committed
[palm] set max out tokens
1 parent d08b08d commit fd8018e

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

llm4j-palm/src/main/java/org/llm4j/palm/request/EmbedRequestFactory.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ public class EmbedRequestFactory {
1111
*/
1212
private String modelId;
1313
private String text;
14-
private PaLMRequestParameters parameters;
1514

1615
public EmbedRequestFactory withText(String text) {
1716
this.text = text;

llm4j-palm/src/main/java/org/llm4j/palm/request/TextRequestFactory.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ public GenerateTextRequest build() {
3434
.setPrompt(prompt)
3535
.setTemperature(parameters.temperature.floatValue())
3636
.setCandidateCount(parameters.candidateCount)
37-
.setMaxOutputTokens(parameters.maxNewTokens)
37+
.setMaxOutputTokens(parameters.maxOutputTokens)
3838
.setTopK(parameters.topK)
3939
.setTopP(parameters.topP.floatValue())
4040
.build();

0 commit comments

Comments
 (0)