Update README.md

README.md

@@ -175,33 +175,23 @@ LINCE-ZERO was trained on AWS SageMaker, on ... GPUs in ... instances.
 
 ### Software
 
-More information needed
+We used the following libraries:
+- transformers
+- accelerate
+- peft
+- bitsandbytes
+- einops
 
 # 🌳 Environmental Impact
 
 Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
 
-- **Hardware Type:** More information needed
-- **Hours used:** More information needed
-- **Cloud Provider:** More information needed
-- **Compute Region:** More information needed
-- **Carbon Emitted:** More information needed
-
-# 📝 Citation
-
-There is a paper coming soon! Meanwhile, when using LINCE-ZERO please use the following information to cite:
-
-```markdown
-@article{lince-zero,
-  title={{LINCE-ZERO}: Llm for Instructions from Natural Corpus en Español},
-  author={},
-  year={2023}
-}
-```
-
-# 📧 Contact
+- **Hardware Type:** 1 X A100 - 40 GB
+- **Hours used:** 8
+- **Cloud Provider:** Google
+- **Compute Region:** Europe
+- **Carbon Emitted:** 250W x 10h = 2.5 kWh x 0.57 kg eq. CO2/kWh = 1.42 kg eq. CO2
 
-[[email protected]](mailto:[email protected])
 
 # 🔥 How to Get Started with LINCE-ZERO
 
@@ -213,7 +203,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer
 
 model_id = "clibrain/lince-zero"
 
-model = AutoModelForCausalLM.from_pretrained(model_id).to("cuda")
+model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True).to("cuda")
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 def create_instruction(instruction, input_data=None, context=None):
@@ -248,7 +238,7 @@ def generate(
 ):
     
     prompt = create_instruction(instruction, input, context)
-    print(prompt)
+    print(prompt.replace("### Respuesta:\n", "")
     inputs = tokenizer(prompt, return_tensors="pt")
     input_ids = inputs["input_ids"].to("cuda")
     attention_mask = inputs["attention_mask"].to("cuda")
@@ -275,4 +265,20 @@ def generate(
 
 instruction = "Dame una lista de lugares a visitar en España."
 print(generate(instruction))
-```
+```
+
+# 📝 Citation
+
+There is a paper coming soon! Meanwhile, when using LINCE-ZERO please use the following information to cite:
+
+```markdown
+@article{lince-zero,
+  title={{LINCE-ZERO}: Llm for Instructions from Natural Corpus en Español},
+  author={},
+  year={2023}
+}
+```
+
+# 📧 Contact
+
+[[email protected]](mailto:[email protected])