chaowenguo0101

### 需求描述 | Problem Statement how to specify the length of video? ### 建议的解决方案 | Proposed Solution how to specify the length of video?

### 需求描述 | Problem Statement how to improve resolution in the final video. which source code I need to change? ### 建议的解决方案 | Proposed Solution how to improve resolution in...

device_map can not use xla device such as tpu

1

### Describe the bug ```import torch, pathlib, builtins, transformers, diffusers, huggingface_hub model_id = 'Wan-AI/Wan2.1-T2V-14B-Diffusers' model_dir = huggingface_hub.snapshot_download(model_id) pipe = diffusers.WanPipeline.from_pretrained(model_dir, vae=diffusers.AutoencoderKLWan.from_pretrained(model_dir, subfolder='vae', torch_dtype=torch.float32), torch_dtype=torch.bfloat16, device_map='balanced') pipe.scheduler = diffusers.UniPCMultistepScheduler(prediction_type='flow_prediction', use_flow_sigmas=True, num_train_timesteps=1000,...