DistillationSpec

public abstract class DistillationSpec extends JsonSerializable

Distillation tuning spec for tuning.

Constructors

Link copied to clipboard
public void DistillationSpec()

Types

Link copied to clipboard
public abstract class Builder
Builder for DistillationSpec.

Properties

Link copied to clipboard
public final static String MAX_READ_LENGTH_PROPERTY
System property to override the default max JSON string length (20MB) in read constraints.

Functions

Link copied to clipboard
public abstract Optional<String> baseTeacherModel()
The base teacher model that is being distilled.
Link copied to clipboard
Instantiates a builder for DistillationSpec.
Link copied to clipboard
public static DistillationSpec fromJson(String jsonString)
Deserializes a JSON string to a DistillationSpec object.
Link copied to clipboard
@InternalApi()
public static T fromJsonNode<T extends JsonSerializable>(JsonNode jsonNode, Class<T> clazz)
Deserializes a JsonNode to an object of the given type.
Link copied to clipboard
@InternalApi()
public static T fromJsonString<T extends JsonSerializable>(String jsonString, Class<T> clazz)
Deserializes a Json string to an object of the given type.
Link copied to clipboard
Link copied to clipboard
public static ObjectMapper objectMapper()
Link copied to clipboard
Deprecated.
Link copied to clipboard
public abstract Optional<String> promptDatasetUri()
The GCS URI of the prompt dataset to use during distillation.
Link copied to clipboard
public static void setMaxReadLength(int maxReadLength)
Overrides the default maximum JSON string length (20MB) for the JSON parser.
Link copied to clipboard
public static JsonNode stringToJsonNode(String string)
Converts a Json string to a JsonNode.
Link copied to clipboard
public abstract Optional<String> studentModel()
The student model that is being tuned, e.g., "google/gemma-2b-1.1-it".
Link copied to clipboard
Creates a builder with the same values as this instance.
Link copied to clipboard
public String toJson()
Serializes the instance to a Json string.
Link copied to clipboard
public static JsonNode toJsonNode(Object object)
Serializes an object to a JsonNode.
Link copied to clipboard
public static String toJsonString(Object object)
Serializes an object to a Json string.
Link copied to clipboard
public abstract Optional<String> trainingDatasetUri()
Deprecated.
Link copied to clipboard
The resource name of the Tuned teacher model.
Link copied to clipboard
public abstract Optional<TuningMode> tuningMode()
Tuning mode for tuning.
Link copied to clipboard
Optional.