You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
- In the folder src/maxdiffusion/models/ltx_video/utils, run:
178
179
```bash
179
180
python convert_torch_weights_to_jax.py --ckpt_path [LOCAL DIRECTORY FOR WEIGHTS] --transformer_config_path ../xora_v1.2-13B-balanced-128.json
@@ -216,7 +217,6 @@ To generate images, run the following command:
216
217
```bash
217
218
python src/maxdiffusion/generate_flux.py src/maxdiffusion/configs/base_flux_schnell.yml jax_cache_dir=/tmp/cache_dir run_name=flux_test output_dir=/tmp/ prompt="photograph of an electronics chip in the shape of a race car with trillium written on its side" per_device_batch_size=1 ici_data_parallelism=1 ici_fsdp_parallelism=-1 offload_encoders=False
218
219
```
219
-
220
220
## Fused Attention for GPU:
221
221
Fused Attention for GPU is supported via TransformerEngine. Installation instructions:
222
222
@@ -333,3 +333,5 @@ This script will automatically format your code with `pyink` and help you identi
333
333
334
334
335
335
The full suite of -end-to end tests is in `tests` and `src/maxdiffusion/tests`. We run them with a nightly cadance.
0 commit comments