Update README.md

README.md

@@ -1,3 +1,12 @@
+---
+language: 
+  - en
+datasets:
+- super_glue
+---
+
+see [this repo](https://huggingface.co/crumb/gpt-j-6b-shakespeare) for more information
+
 ```python
 from IPython import display 
 !pip install transformers==4.14.1 -q
@@ -6,7 +15,7 @@ from IPython import display
 
 import transformers_8bit
 
-model, tokenizer, config = transformers_8bit.gptj("crumb/gpt-j-6b-finetune-super-glue-test", device='cuda')
+model, tokenizer, config = transformers_8bit.gptj("crumb/gpt-j-6b-finetune-super-glue", device='cuda')
 
 prompt = tokenizer("<QUERY> If birds are in group B, and snakes are in group A, what group are pythons in? <RESPONSE>", return_tensors='pt')
 prompt = {key: value.to('cuda') for key, value in prompt.items()}