wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-16 17:42:58 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
E-Anlia 2025-12-12 16:01:57 +08:00 committed by GitHub
parent 4c08fd2150
commit 5fcd6c5f79
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1557 additions and 1557 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
comfy/supported_models.py
View File

@ -1036,20 +1036,20 @@ class ZImage(Lumina2):
return supported_models_base.ClipTarget(comfy.text_encoders.z_image.ZImageTokenizer, comfy.text_encoders.z_image.te(**hunyuan_detect))
class NewBieImageModel(supported_models_base.BASE):
unet_config = {
"image_model": "NewBieImage",
"model_type": "newbie_dit",
}
sampling_settings = {
"multiplier": 1.0,
"shift": 6.0,
}
memory_usage_factor = 1.5
unet_extra_config = {}
latent_format = latent_formats.Flux
supported_inference_dtypes = [torch.bfloat16, torch.float16, torch.float32]
vae_key_prefix = ["vae."]
text_encoder_key_prefix = ["text_encoders."]
unet_config = {
"image_model": "NewBieImage",
"model_type": "newbie_dit",
}
sampling_settings = {
"multiplier": 1.0,
"shift": 6.0,
}
memory_usage_factor = 1.5
unet_extra_config = {}
latent_format = latent_formats.Flux
supported_inference_dtypes = [torch.bfloat16, torch.float16, torch.float32]
vae_key_prefix = ["vae."]
text_encoder_key_prefix = ["text_encoders."]
def get_model(self, state_dict, prefix="", device=None):
out = model_base.NewBieImage(self, device=device)