mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-08-21 20:46:50 +08:00
20 lines
913 B
Python
20 lines
913 B
Python
|
# Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
|
|||
|
|
import torch
|
|||
|
|
from easydict import EasyDict
|
|||
|
|
|
|||
|
|
#------------------------ Wan shared config ------------------------#
|
|||
|
|
wan_shared_cfg = EasyDict()
|
|||
|
|
|
|||
|
|
# t5
|
|||
|
|
wan_shared_cfg.t5_model = 'umt5_xxl'
|
|||
|
|
wan_shared_cfg.t5_dtype = torch.float32
|
|||
|
|
wan_shared_cfg.text_len = 512
|
|||
|
|
|
|||
|
|
# transformer
|
|||
|
|
wan_shared_cfg.param_dtype = torch.float32
|
|||
|
|
|
|||
|
|
# inference
|
|||
|
|
wan_shared_cfg.num_train_timesteps = 1000
|
|||
|
|
wan_shared_cfg.sample_fps = 16
|
|||
|
|
wan_shared_cfg.sample_neg_prompt = '色调艳丽，过曝，静态，细节模糊不清，字幕，风格，作品，画作，画面，静止，整体发灰，最差质量，低质量，JPEG压缩残留，丑陋的，残缺的，多余的手指，画得不好的手部，画得不好的脸部，畸形的，毁容的，形态畸形的肢体，手指融合，静止不动的画面，杂乱的背景，三条腿，背景人很多，倒着走'
|