Update README.md

README.md

@@ -8,7 +8,7 @@ language:
 pipeline_tag: text-generation
 ---
 
-# Model Card for Model ID
+# Model Card for LegoGPT
 
 <!-- Provide a quick summary of what the model is/does. -->
 
@@ -35,8 +35,8 @@ pipeline_tag: text-generation
 <!-- Provide the basic links for the model. -->
 
 - **Repository:** [AvaLovelace1/LegoGPT](https://github.com/AvaLovelace1/LegoGPT)
-- **Paper [optional]:** [More Information Needed]
-- **Demo [optional]:** [More Information Needed]
+- **Paper [optional]:** TBA
+- **Demo [optional]:** TBA
 
 ## Uses
 
@@ -82,9 +82,7 @@ Use the code below to get started with the model.
 
 ### Training Data
 
-<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
-
-[More Information Needed]
+LegoGPT was trained using [StableText2Lego](https://huggingface.co/datasets/AvaLovelace/StableText2Lego), a dataset of 47k LEGO structures.
 
 ### Training Procedure
 
@@ -132,72 +130,28 @@ The model was fine-tuned using LoRA applied to the `q_proj` and `v_proj` matrice
 #### Summary
 
 
-
-## Model Examination [optional]
-
-<!-- Relevant interpretability work for the model goes here -->
-
-[More Information Needed]
-
 ## Environmental Impact
 
-<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
-
-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]
+- **Hardware Type:** 8x NVIDIA RTX A6000 (48 GB)
+- **Hours used:** 0.5
 - **Carbon Emitted:** [More Information Needed]
 
-## Technical Specifications [optional]
-
-### Model Architecture and Objective
-
-[More Information Needed]
-
-### Compute Infrastructure
-
-[More Information Needed]
-
-#### Hardware
-
-[More Information Needed]
-
-#### Software
-
-[More Information Needed]
-
-## Citation [optional]
+## Citation
 
 <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
 
 **BibTeX:**
 
-[More Information Needed]
+TBA
 
 **APA:**
 
-[More Information Needed]
-
-## Glossary [optional]
-
-<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
-
-[More Information Needed]
-
-## More Information [optional]
-
-[More Information Needed]
-
-## Model Card Authors [optional]
-
-[More Information Needed]
+TBA
 
 ## Model Card Contact
 
-[More Information Needed]
+Ava Pun ([email protected])
+
 ### Framework versions
 
 - PEFT 0.15.0