105 lines
3.5 KiB
Markdown
105 lines
3.5 KiB
Markdown
|
# Image generation
|
||
|
|
|
||
|
|
In Agents-Flex, the ability to generate images through AI is built-in.
|
||
|
|
|
||
|
|
## Large model support
|
||
|
|
Agents-Flex image generation models support the following:
|
||
|
|
|
||
|
|
| Large Language Model Name | Support Status | Description |
|
||
|
|
|--------------------------------------------|--------|-------|
|
||
|
|
| Openai | ✅ Supported | - |
|
||
|
|
| Stability | ✅ Supported | - |
|
||
|
|
| GiteeAI - stable-diffusion-3-medium | ✅ Supported | - |
|
||
|
|
| GiteeAI - FLUX.1-schnell | ✅ Supported | - |
|
||
|
|
| GiteeAI - stable-diffusion-xl-base-1.0 | ✅ Supported | - |
|
||
|
|
| GiteeAI - Kolors | ✅ Supported | - |
|
||
|
|
| SiliconFlow - Flux.1-schnell | ✅ Supported | - |
|
||
|
|
| SiliconFlow - Stable Diffusion 3 | ✅ Supported | - |
|
||
|
|
| SiliconFlow - Stable Diffusion XL | ✅ Supported | - |
|
||
|
|
| SiliconFlow - Stable Diffusion 2.1 | ✅ Supported | - |
|
||
|
|
| SiliconFlow - Stable Diffusion Turbo | ✅ Supported | - |
|
||
|
|
| SiliconFlow - Stable Diffusion XL Turbo | ✅ Supported | - |
|
||
|
|
| SiliconFlow - Stable Diffusion XL Lighting | ✅ Supported | - |
|
||
|
|
| More |Planning... | Looking forward to PR |
|
||
|
|
|
||
|
|
## Example Code
|
||
|
|
|
||
|
|
```java
|
||
|
|
@Test
|
||
|
|
public void testGenImage(){
|
||
|
|
GiteeImageModelConfig config = new GiteeImageModelConfig();
|
||
|
|
config.setApiKey("****");
|
||
|
|
|
||
|
|
//Step 1: Create an ImageModel
|
||
|
|
ImageModel imageModel = new GiteeImageModel(config);
|
||
|
|
|
||
|
|
//Step 2: Create pictures to generate prompt words and parameters
|
||
|
|
GenerateImageRequest request = new GenerateImageRequest();
|
||
|
|
request.setPrompt("A cute little tiger standing in the high-speed train");
|
||
|
|
request.setSize(1024, 1024);
|
||
|
|
|
||
|
|
//Step 3: Generate images through large models
|
||
|
|
ImageResponse generate = imageModel.generate(request);
|
||
|
|
System.out.println(generate);
|
||
|
|
|
||
|
|
int index = 0;
|
||
|
|
for (Image image : generate.getImages()) {
|
||
|
|
//Step 4: Save the image locally
|
||
|
|
image.writeToFile(new File("/image-path/"+(index++)+".jpg"));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
```
|
||
|
|
|
||
|
|
或者使用 OpenAI ImageModel
|
||
|
|
|
||
|
|
```java 5-7
|
||
|
|
@Test
|
||
|
|
public void testGenImage(){
|
||
|
|
|
||
|
|
//Or use OpenAI ImageModel
|
||
|
|
OpenAIImageModelConfig config = new OpenAIImageModelConfig();
|
||
|
|
config.setApiKey("sk-5gqOclb****");
|
||
|
|
ImageModel imageModel = new OpenAIImageModel(config);
|
||
|
|
|
||
|
|
|
||
|
|
GenerateImageRequest request = new GenerateImageRequest();
|
||
|
|
request.setPrompt("A cute little tiger standing in the high-speed train");
|
||
|
|
request.setSize(1024, 1024);
|
||
|
|
|
||
|
|
ImageResponse generate = imageModel.generate(request);
|
||
|
|
System.out.println(generate);
|
||
|
|
|
||
|
|
int index = 0;
|
||
|
|
for (Image image : generate.getImages()) {
|
||
|
|
image.writeToFile(new File("/image-path/"+(index++)+".jpg"));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
```
|
||
|
|
|
||
|
|
Or use SiliconFlowImageModel
|
||
|
|
|
||
|
|
```java 5-8
|
||
|
|
@Test
|
||
|
|
public void testGenImage(){
|
||
|
|
|
||
|
|
//Or use SiliconFlowImageModel
|
||
|
|
SiliconflowImageModelConfig config = new SiliconflowImageModelConfig();
|
||
|
|
config.setModel(SiliconflowImageModels.Stable_Diffusion_XL);
|
||
|
|
config.setApiKey("sk-****");
|
||
|
|
ImageModel imageModel = new OpenAIImageModel(config);
|
||
|
|
|
||
|
|
|
||
|
|
GenerateImageRequest request = new GenerateImageRequest();
|
||
|
|
request.setPrompt("A cute little tiger standing in the high-speed train");
|
||
|
|
request.setSize(1024, 1024);
|
||
|
|
|
||
|
|
ImageResponse generate = imageModel.generate(request);
|
||
|
|
System.out.println(generate);
|
||
|
|
|
||
|
|
int index = 0;
|
||
|
|
for (Image image : generate.getImages()) {
|
||
|
|
image.writeToFile(new File("/image-path/"+(index++)+".jpg"));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
```
|