Update README.md

README.md

@@ -14,58 +14,24 @@ datasets:
 
 # MeteoLibre Rectified Flow Model
 
-This is a repo with differents models used for doing weather forecasting :
-- epoch_126_mtg_meteofrance_.safetensors (model with sat + ground station) : config is model_v0_mtg_meteofrance
 
 ## Model Description
 
 - **Model type**: Rectified Flow Diffusion Model
 - **Architecture**: 3D U-Net with FiLM conditioning
-- **Input**: Meteorological data patches (12 channels, 3D spatio-temporal)
+- **Input**: Meteorological data patches (12 channels + 1 lightning channels, 3D spatio-temporal)
 - **Output**: Generated weather forecast data
 - **Training data**: MeteoLibre meteorological dataset
 - **Language(s)**: Python
 - **License**: MIT
 
-## Intended Use
-
-This model is designed for:
-- Weather pattern generation and forecasting
-- Meteorological data augmentation
-- Research in atmospheric science and weather prediction
-- Educational purposes in machine learning for climate modeling
-
 ## Training
 
 The model was trained using:
 - **Framework**: PyTorch with Hugging Face Accelerate
-- **Optimizer**: Adam (lr=5e-4)
+- **Optimizer**: Adam (lr=5e-4) OR SOAP
 - **Batch size**: 64
 - **Epochs**: 200
 - **Precision**: Mixed precision (bf16)
 - **Distributed training**: Multi-GPU support
 
-## Ethical Considerations
-
-- Weather forecasting models should be used responsibly
-- Consider environmental impact of computational requirements
-- Validate predictions against ground truth data
-- Not intended for critical decision-making without human oversight
-
-## Citation
-
-If you use this model in your research, please cite:
-
-```bibtex
-@misc{meteolibre-rectified-flow,
-  title={MeteoLibre Rectified Flow Weather Forecasting Model},
-  author={MeteoLibre Development Team},
-  year={2025},
-  publisher={Hugging Face},
-  url={https://huggingface.co/meteolibre-dev/meteolibre-rectified-flow}
-}
-```
-
-## Contact
-
-For questions or issues, please open an issue on the [MeteoLibre GitHub repository](https://github.com/meteolibre-dev/meteolibre_model).