git/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2026-03-13 18:30:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

Remove the ParamScheduler for z-images

Browse Source 
Fixed the DEFAULT_TOKENIZER_SOURCE to Qwen/Qwen3-4B
This commit is contained in:
Alexander Eichhorn 2025-12-13 04:23:34 +01:00
parent f9605e18a0
commit 3ee24cbdde
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/backend/model_manager/load/model_loaders/z_image.py
View File

@ -379,7 +379,8 @@ class Qwen3EncoderCheckpointLoader(ModelLoader):
"""Class to load single-file Qwen3 Encoder models for Z-Image (safetensors format)."""
# Default HuggingFace model to load tokenizer from when using single-file Qwen3 encoder
DEFAULT_TOKENIZER_SOURCE = "Qwen/Qwen2.5-3B"
# Must be Qwen3 (not Qwen2.5) to match Z-Image's text encoder architecture and special tokens
DEFAULT_TOKENIZER_SOURCE = "Qwen/Qwen3-4B"
def _load_model(
self,
@ -543,7 +544,8 @@ class Qwen3EncoderGGUFLoader(ModelLoader):
"""Class to load GGUF-quantized Qwen3 Encoder models for Z-Image."""
# Default HuggingFace model to load tokenizer from when using GGUF Qwen3 encoder
DEFAULT_TOKENIZER_SOURCE = "Qwen/Qwen2.5-3B"
# Must be Qwen3 (not Qwen2.5) to match Z-Image's text encoder architecture and special tokens
DEFAULT_TOKENIZER_SOURCE = "Qwen/Qwen3-4B"
def _load_model(
self,

5
invokeai/frontend/web/src/features/settingsAccordions/components/GenerationSettingsAccordion/GenerationSettingsAccordion.tsx
View File

@ -4,7 +4,7 @@ import { EMPTY_ARRAY } from 'app/store/constants';
import { createMemoizedSelector } from 'app/store/createMemoizedSelector';
import { useAppSelector } from 'app/store/storeHooks';
import { selectLoRAsSlice } from 'features/controlLayers/store/lorasSlice';
import { selectIsCogView4, selectIsFLUX, selectIsSD3 } from 'features/controlLayers/store/paramsSlice';
import { selectIsCogView4, selectIsFLUX, selectIsSD3, selectIsZImage } from 'features/controlLayers/store/paramsSlice';
import { LoRAList } from 'features/lora/components/LoRAList';
import LoRASelect from 'features/lora/components/LoRASelect';
import ParamCFGScale from 'features/parameters/components/Core/ParamCFGScale';
@ -29,6 +29,7 @@ export const GenerationSettingsAccordion = memo(() => {
const isFLUX = useAppSelector(selectIsFLUX);
const isSD3 = useAppSelector(selectIsSD3);
const isCogView4 = useAppSelector(selectIsCogView4);
const isZImage = useAppSelector(selectIsZImage);
const selectBadges = useMemo(
() =>
@ -66,7 +67,7 @@ export const GenerationSettingsAccordion = memo(() => {
<Expander label={t('accordions.advanced.options')} isOpen={isOpenExpander} onToggle={onToggleExpander}>
<Flex gap={4} flexDir="column" pb={4}>
<FormControlGroup formLabelProps={formLabelProps}>
{!isFLUX && !isSD3 && !isCogView4 && <ParamScheduler />}
{!isFLUX && !isSD3 && !isCogView4 && !isZImage && <ParamScheduler />}
<ParamSteps />
{isFLUX && modelConfig && !isFluxFillMainModelModelConfig(modelConfig) && <ParamGuidance />}
{!isFLUX && <ParamCFGScale />}