Update README.md
Browse files
README.md
CHANGED
@@ -276,11 +276,25 @@ Carbon emissions can be estimated using the [Machine Learning Impact calculator]
|
|
276 |
|
277 |
#### Hardware
|
278 |
|
279 |
-
|
280 |
|
281 |
#### Software
|
282 |
|
283 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
284 |
|
285 |
## Citation
|
286 |
|
@@ -297,4 +311,4 @@ Carbon emissions can be estimated using the [Machine Learning Impact calculator]
|
|
297 |
|
298 |
## Model Card Contact
|
299 |
|
300 |
-
- contacto.iic@iic.uam.es
|
|
|
276 |
|
277 |
#### Hardware
|
278 |
|
279 |
+
GPU NVIDIA A100 with Tensor Core.
|
280 |
|
281 |
#### Software
|
282 |
|
283 |
+
`Ubuntu 22.04.5 LTS` and the following requirements:
|
284 |
+
|
285 |
+
```
|
286 |
+
python=3.11
|
287 |
+
flash_attn>=2.5.8
|
288 |
+
datasets
|
289 |
+
numpy
|
290 |
+
tabulate
|
291 |
+
openpyxl
|
292 |
+
trl
|
293 |
+
peft
|
294 |
+
bitsandbytes
|
295 |
+
huggingface_hub
|
296 |
+
tensorboard
|
297 |
+
```
|
298 |
|
299 |
## Citation
|
300 |
|
|
|
311 |
|
312 |
## Model Card Contact
|
313 |
|
314 |
+
- [contacto.iic@iic.uam.es](contacto.iic@iic.uam.es).
|