Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
7e8ec87
1
Parent(s):
35b9a3d
Update app.py
Browse files
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=
|
| 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 |
|