Hugging Face's logo

Spaces:
multimodalart
/
self-forcing
Running on Zero

App Files Community
1
multimodalart HF Staff commited on
Commit
e363e69
·
verified ·
1 Parent(s): f29396c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ from demo_utils.constant import ZERO_VAE_CACHE
40
  from demo_utils.vae_block3 import VAEDecoderWrapper
41
  from utils.wan_wrapper import WanDiffusionWrapper, WanTextEncoder
42
 
43
- from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
44
 
45
  device = "cuda" if torch.cuda.is_available() else "cpu"
46
 
 
40
  from demo_utils.vae_block3 import VAEDecoderWrapper
41
  from utils.wan_wrapper import WanDiffusionWrapper, WanTextEncoder
42
 
43
+ from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
44
 
45
  device = "cuda" if torch.cuda.is_available() else "cpu"
46