stable-diffusion-rpc/Common/host.proto

/**
 * Host management for the Stable Diffusion RPC service.
 */
syntax = "proto3";
package gay.pizza.stable.diffusion;
import "shared.proto";

/**
 * Utilize a prefix of 'Sd' for Swift.
 */
option swift_prefix = "Sd";
option java_multiple_files = true;

/**
 * Represents a request to load a model into a specified compute unit.
 */
message LoadModelRequest {
  /**
   * The model name to load onto the compute unit.
   */
  string model_name = 1;

  /**
   * The compute units to load the model onto.
   */
  ComputeUnits compute_units = 2;
}

/**
 * Represents a response to loading a model.
 */
message LoadModelResponse {}

/**
 * The model service, for management and loading of models.
 */
service HostModelService {
  /**
   * Loads a model onto a compute unit.
   */
  rpc LoadModel(LoadModelRequest) returns (LoadModelResponse);
}