nyarunyarunya commited on
Commit
8ab81be
·
verified ·
1 Parent(s): 7ba1e43

Update wan/modules/vae.py

Browse files
Files changed (1) hide show
  1. wan/modules/vae.py +1 -1
wan/modules/vae.py CHANGED
@@ -1,5 +1,5 @@
1
  # Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
2
- import logging
3
 
4
  import torch
5
  import torch.cuda.amp as amp
 
1
  # Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
2
+ import logging,spaces
3
 
4
  import torch
5
  import torch.cuda.amp as amp