Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ public class LlmModule {

public static final int MODEL_TYPE_TEXT = 1;
public static final int MODEL_TYPE_TEXT_VISION = 2;
public static final int MODEL_TYPE_MULTIMODAL = 2;

private final HybridData mHybridData;
private static final int DEFAULT_SEQ_LEN = 128;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,9 @@ private LlmModuleConfig(Builder builder) {
/** Model type constant for text-and-vision multimodal models. */
public static final int MODEL_TYPE_TEXT_VISION = 2;

/** Model type constant for generic multimodal models. */
public static final int MODEL_TYPE_MULTIMODAL = 2;

/**
* Creates a new Builder instance for constructing LlmModuleConfig objects.
*
Expand Down Expand Up @@ -141,7 +144,7 @@ public Builder dataPath(String dataPath) {
/**
* Sets the model type (text-only or multimodal).
*
* @param modelType One of MODEL_TYPE_TEXT or MODEL_TYPE_TEXT_VISION
* @param modelType One of MODEL_TYPE_TEXT, MODEL_TYPE_TEXT_VISION, MODEL_TYPE_MULTIMODAL
* @return This builder instance for method chaining
*/
public Builder modelType(int modelType) {
Expand Down
Loading