AI-Hypercomputer / AI-Hypercomputer/JetStream
Clean up Model Conversion Script
Ouverte
- Langage dominant
- Python
- Étoiles
- 457
- Forks
- 67
- Métriques de merge des PR
- Aucune PR mergée en 30 j
Description
Currently the model conversion script will [create a bucket](https://github.com/google/JetStream/blob/main/jetstream/tools/maxtext/model_ckpt_conversion.sh#L36) `export MODEL_BUCKET=gs://${USER}-maxtext`. However, it may be the case that the `gs://${USER}-maxtext` path already exists, which I imagine would break the script.
Solution: Be able to read in a few more arguments `MODEL_BUCKET` and `BASE_OUTPUT_DIRECTORY`. We should also delete references to `DATASET_PATH`.
Guide de contribution
Ouvrir le guide de contribution
Évaluation
Cette issue n'a pas encore été évaluée.