Skip to content

Commit 19b34db

Browse files
committed
⚡ Flux2 resolution, 🔥 Llama3PredictionMapper
1 parent cda8bb5 commit 19b34db

2 files changed

Lines changed: 2 additions & 129 deletions

File tree

spring-boot-starter-replicate/src/main/java/io/graversen/replicate/flux/Flux2PredictionMapper.java

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
import io.graversen.replicate.service.CreateImagePrediction2;
66
import lombok.NonNull;
77
import lombok.RequiredArgsConstructor;
8-
import lombok.SneakyThrows;
98
import org.springframework.stereotype.Component;
109

1110
import java.net.URI;
@@ -135,7 +134,7 @@ Function<CreateImagePrediction2, BlackForestLabsFlux2ProInput> mapFluxProInput()
135134
null,
136135
null,
137136
createImagePrediction.getPrompt(),
138-
null,
137+
"1 MP",
139138
mapAspectRatio(createImagePrediction),
140139
mapInputImages(createImagePrediction),
141140
mapOutputFormat(createImagePrediction),
@@ -156,9 +155,7 @@ private Boolean mapDisableSafetyChecker(@NonNull CreateImagePrediction2 createIm
156155

157156
private Integer mapSafetyTolerance(@NonNull CreateImagePrediction2 createImagePrediction) {
158157
if (createImagePrediction.getModerationLevel() != null) {
159-
return createImagePrediction.getModerationLevel() == ModerationLevels.HIGH
160-
? 2
161-
: 5;
158+
return createImagePrediction.getModerationLevel() == ModerationLevels.HIGH ? 2 : 5;
162159
} else {
163160
return 5;
164161
}

spring-boot-starter-replicate/src/main/java/io/graversen/replicate/llama3/Llama3PredictionMapper.java

Lines changed: 0 additions & 124 deletions
This file was deleted.

0 commit comments

Comments
 (0)