This is not an officially supported Google product. This project is not eligible for the Google Open Source Software Vulnerability Rewards Program.

This is a preview version of Google GenAI custom nodes for ComfyUI.

ComfyUI custom nodes for Google GenMedia

This repository contains ComfyUI custom nodes for the following Google GenMedia models: - Gemini (2.5, 3.1, 3.5) - Gemini 2.5 Flash Image (Nano Banana 🍌) - Gemini 3.1 Flash Image (Nano Banana 2 🍌) - Gemini 3 Pro Image (Nano Banana Pro 🍌) - Imagen3 - Imagen4 - Lyria2 - Veo2 - Veo3.1 - Virtual-try-on

All of these nodes need a Google Cloud project and region as input and uses the Vertex AI APIs to access Google GenMedia models. If you are running ComfyUI on GKE as described in ComfyUI on GKE guide, the Vertex AI APIs will be authenticated with GKE workload identity and no additional action is required form your end.

If you are running your own(local) ComfyUI installation, you will need to install the custom nodes and authenticate to Google Cloud as an user or a Service Account. See installation section for details.

Pricing

All the custom nodes provided with this repo use Vertex AI APIs to generate content and hence, Vertex AI pricing is applied to each API call. See Vertex AI Pricing for the model that you want to use.

Note : If you store the content in Google Cloud Storage using these custom nodes, refer to Google Cloud Storage pricing.

Installing the Google Generative Media custom nodes

ComfyUI on GKE

If you want to run ComfyUI on GKE and use these custom nodes, follow the ComfyUI on GKE guide that will spin up ComfyUI on GKE with these custom nodes installed and ready to use.

ComfyUI running locally

If you are running ComfyUI locally or at a place other than GCP and want to use these custom nodes, you need to perform the following steps to install them:

1. Download these custom nodes in one of the following ways:

   • Using the ComfyUI Manager

   • Go to ComfyUI Manager --> Custom Node Manager --> search for comfyui-google-genmedia-custom-nodes. Click Install.

   • Manual installation using git

   • Open the terminal on the machine that is running ComfyUI and change to the directory where ComfyUI is running from. ```sh cd

     E.g if your ComfyUI is running from "/home/user/username/ComfyUI", replace <PATH_TO_COMFYUI_INSTALLATION> with "/home/user/username/"
     ```
        - Change to the custom_nodes folder.
     

     sh cd ComfyUI/custom_nodes

   • Clone this repository in the custom_nodes folder: sh git clone https://github.com/GoogleCloudPlatform/comfyui-google-genmedia-custom-nodes

   • Install python packages:
     1. If you are running ComfyUI in a python virtual environment, activate it before installing the dependencies. shREADME.md source <PATH_TO_PYTHON_VIRTUAL_ENVIRONMENT>/bin/activate
     2. Install python packages sh pip install -r comfyui-google-genmedia-custom-nodes/requirements.txt

2. Now that the custom nodes are installed, set up authentication with Google Cloud. This is needed because the custom nodes will use Vertex AI APIs in the given Google Cloud project and region. Choose one of the following authentication methods:

   • Authenticate local ComfyUI with Google Cloud user
   • Authenticate local ComfyUI with Google Cloud Service Accountt

3. Re-start ComfyUI via ComfyUI manager or manually. Verify there are no failures in the logs and the log says that google-genmedia custom node has been loaded successfully, similar to the following message: sh Import times for custom nodes: ..... ..... ..... 2.8 seconds: /XXX/XXXX/ComfyUI/custom_nodes/comfyui-google-genmedia-custom-nodes

4. You are all set! Start using the custom nodes. Right click on ComfyUI, Click the 'Add node' menu and you should see a category named Google AI with custom nodes for Google’s genmedia models as shown in the image below. You will need to pass a GCP project id and GCP region as input parameter to every node if you are using the custom nodes from your local ComfyUI installation.
5. Optional step : If you are not familiar with Google's GenMedia custom nodes, we provide some sample workflows with this repo that shows how to use the nodes. You can download and use the sample workflows by following the steps in Sample Workflows section. For full set of features and input parameters to each node, see Node features section for more details.

Authenticate local ComfyUI with Google Cloud user

• On the machine where you are running ComfyUI, install gcloud cli using the documentation https://docs.cloud.google.com/sdk/docs/install
• Authenticate with the following command. Enter your Google Cloud credentials when prompted: sh gcloud auth application-default login

Authenticate local ComfyUI with Google Cloud Service Account

• On the machine where you are running ComfyUI, install gcloud cli using the documentation https://docs.cloud.google.com/sdk/docs/install
• Authenticate with the following command. Enter your Google Cloud credentials when prompted: sh gcloud auth login

• Set shell variables ``sh export PROJECT_ID="<YOUR_PROJECT_ID>" export SERVICE_ACCOUNT_ID="<YOUR_SERVICE_ACCOUNT_ID>" export USER_EMAIL=gcloud auth list --filter=status:ACTIVE --format="value(account)"`

  Replace with the GCP project id you plan to use Replace with the name of the service account you will use to access Google Cloud with ComfyUI. It will be created in the next step. - Set Google Cloud project.sh gcloud config set project $PROJECT_ID - Create a new service account:sh gcloud iam service-accounts create $SERVICE_ACCOUNT_ID \ --display-name="ComfyUI Service Account" \ --description="Service Account to use ComfyUI" - Grant permissions to the service account:sh gcloud projects add-iam-policy-binding $PROJECT_ID \ --member="serviceAccount:${SERVICE_ACCOUNT_ID}@${PROJECT_ID}.iam.gserviceaccount.com" \ --role="roles/storage.objectUser" --condition=None

  gcloud projects add-iam-policy-binding $PROJECT_ID \ --member="serviceAccount:${SERVICE_ACCOUNT_ID}@${PROJECT_ID}.iam.gserviceaccount.com" \ --role="roles/aiplatform.user" --condition=None - Grant the Service Account Token Creator role to your user on the service account. This role lets your user account to impersonate the Service Account created in previous steps:sh gcloud iam service-accounts add-iam-policy-binding \ ${SERVICE_ACCOUNT_ID}@${PROJECT_ID}.iam.gserviceaccount.com \ --member="user:${USER_EMAIL}" \ --role="roles/iam.serviceAccountTokenCreator" - Now, set yourself to impersonate the service accountsh gcloud auth application-default login --impersonate-service-account=${SERVICE_ACCOUNT_ID}@${PROJECT_ID}.iam.gserviceaccount.com ```

Node features

Gemini 2.5 Custom Node

Category: Google AI/Gemini

This node integrates the Gemini 2.5 into ComfyUI, supporting text generation only with multimodal inputs (image, video, audio).

Required Inputs

Optional Inputs

Outputs

Sample

Gemini 3.1 Custom Node

Category: Google AI/Gemini

This node integrates the Gemini 3.1 into ComfyUI, supporting text generation with multimodal inputs (image, video, audio).

Required Inputs

Optional Inputs

Outputs