Skip to content

[Gemma3] compile at generation time ✨ #37433

Closed
gante wants to merge 4 commits intohuggingface:mainfrom
gante:gemma3_compile
Closed

[Gemma3] compile at generation time ✨ #37433
gante wants to merge 4 commits intohuggingface:mainfrom
gante:gemma3_compile

Conversation

@gante
Copy link
Contributor

@gante gante commented Apr 10, 2025

What does this PR do?

WIP, needs code cleanup, some history, and a PR description

Enables torch.compile for Gemma3 (and fixes it for other models using HybridCache like Gemma2)

@gante gante changed the title [Gemma3] Gemma3 + compile at generation time ✨ [Gemma3] compile at generation time ✨ Apr 11, 2025
@gante
Copy link
Contributor Author

gante commented Apr 11, 2025

found an even cleaner way of doing it 🙏

@gante gante closed this Apr 11, 2025
@gante gante mentioned this pull request Apr 11, 2025
2 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant