VertexAI 设置 GenerationConfig

Question

我已经更改了版本

VertexAI

GenerateContentResponse response = null;
            GenerativeModel model = new GenerativeModel(MODEL_NAME, vertexAI);

但是现在我无法设置

model.setGenerationConfig(GenerationConfig.newBuilder().build());

Cannot resolve method 'setGenerationConfig' in 'GenerativeModel'

Answer 1

您可以将

GenerationConfig

传递给

GenerativeModel

构造函数：

GenerationConfig generationConfig =
              GenerationConfig.newBuilder()
                  .setMaxOutputTokens(100)
                  .setTemperature(0.4F)
                  .setTopK(32)
                  .setTopP(1)
                  .build();

GenerativeModel model = new GenerativeModel(MODEL_NAME, generationConfig, vertexAI);

也请参阅此示例此处。