Skip to content

Commit 78fdf62

Browse files
authored
Fix Z Image model default dtype (#216)
1 parent 63bfbb0 commit 78fdf62

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

diffsynth_engine/configs/pipeline.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -301,11 +301,11 @@ class HunyuanPipelineConfig(BaseConfig):
301301
@dataclass
302302
class ZImagePipelineConfig(AttentionConfig, OptimizationConfig, ParallelConfig, BaseConfig):
303303
model_path: str | os.PathLike | List[str | os.PathLike]
304-
model_dtype: torch.dtype = torch.float16
304+
model_dtype: torch.dtype = torch.bfloat16
305305
vae_path: Optional[str | os.PathLike | List[str | os.PathLike]] = None
306-
vae_dtype: torch.dtype = torch.float16
306+
vae_dtype: torch.dtype = torch.bfloat16
307307
encoder_path: Optional[str | os.PathLike | List[str | os.PathLike]] = None
308-
encoder_dtype: torch.dtype = torch.float16
308+
encoder_dtype: torch.dtype = torch.bfloat16
309309

310310
@classmethod
311311
def basic_config(

0 commit comments

Comments
 (0)