weiyuchoumou526 commited on
Commit
7e8ec87
·
1 Parent(s): 35b9a3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ tokenizer = AutoTokenizer.from_pretrained(tokenizer_path)
65
  text_encoder = WanT5EncoderModel.from_pretrained(
66
  text_encoder_path,
67
  additional_kwargs=OmegaConf.to_container(config['text_encoder_kwargs']),
68
- low_cpu_mem_usage=True,
69
  torch_dtype=torch.bfloat16
70
  )
71
 
 
65
  text_encoder = WanT5EncoderModel.from_pretrained(
66
  text_encoder_path,
67
  additional_kwargs=OmegaConf.to_container(config['text_encoder_kwargs']),
68
+ low_cpu_mem_usage=False,
69
  torch_dtype=torch.bfloat16
70
  )
71