Update README.md

by GopiUppari - opened 4 days ago

base: refs/heads/main

←

from: refs/pr/6

Discussion Files changed

-2

GopiUppari

Google org 4 days ago

The model is running only on the CPU instead of the GPU because the device_map="auto" parameter was not specified in the Gemma3ForConditionalGeneration.from_pretrained function. Adding this parameter will enable automatic GPU allocation.

Update README.md0ba3fdeb

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Ready to merge

This branch is ready to get merged automatically.

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment