Skip to content

Add z-image-turbo as default Replicate model#564

Open
abi wants to merge 1 commit intomainfrom
abi/square-z-image-model
Open

Add z-image-turbo as default Replicate model#564
abi wants to merge 1 commit intomainfrom
abi/square-z-image-model

Conversation

@abi
Copy link
Copy Markdown
Owner

@abi abi commented Feb 26, 2026

Summary

Replaces the hardcoded Flux 2 Klein model with a configurable model selection system. Introduces z-image-turbo as the new default model with 1024×1024 square output, while keeping Flux 2 Klein available as an alternative. The implementation adds a ReplicateImageModel type, model path mapping, and comprehensive test coverage for both models.

Test Plan

  • All existing tests pass (pytest: 68 tests)
  • All type checks pass (pyright: 0 errors, 0 warnings)
  • New tests verify both z-image-turbo and Flux 2 Klein payloads
  • Model selection via module-level constant is testable via monkeypatching

🤖 Generated with Claude Code

Replace hardcoded Flux 2 Klein with configurable model selection. Add z-image-turbo as the new default model with 1024x1024 square output. Keep Flux 2 Klein as an alternative option. Include comprehensive test coverage for both models.

Co-Authored-By: Claude Haiku 4.5 <noreply@anthropic.com>
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