Added JAX weights of galactica-1.3b converted from PyTorch

README.md

@@ -1,3 +1,29 @@
 ---
 license: cc-by-nc-4.0
 ---
+
+JAX weights converted from Torch checkpoint at `facebook/galactica-1.3b`.
+
+```python
+(env) ubuntu@vm:~$ JAX_PLATFORM_NAME=cpu python3
+>>> import jax
+>>> print(jax.devices())   
+[CpuDevice(id=0)]  # Ensure that model weights are loaded into CPU RAM, not accelerator memory.
+>>> from transformers import FlaxOPTForCausalLM
+>>> model = FlaxOPTForCausalLM.from_pretrained("facebook/galactica-1.3b", from_pt=True)
+>>> model.push_to_hub(hf_model_repo)
+```
+
+## Citation and Attribution 
+
+Citation from the original repo is reproduced below as per the cc-by-nc-4.0 licsense.
+
+```bibtex
+@inproceedings{GALACTICA,
+    title={GALACTICA: A Large Language Model for Science},
+    author={Ross Taylor and Marcin Kardas and Guillem Cucurull and Thomas Scialom and Anthony Hartshorn and Elvis Saravia and Andrew Poulton and Viktor Kerkez and Robert Stojnic},
+    year={2022}
+}
+```
+
+> Research supported with Cloud TPUs from Google's TPU Research Cloud (TRC)

config.json

@@ -0,0 +1,32 @@
+{
+  "_name_or_path": "/content/base",
+  "_remove_final_layer_norm": false,
+  "activation_dropout": 0.0,
+  "activation_function": "gelu",
+  "architectures": [
+    "OPTForCausalLM"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 0,
+  "do_layer_norm_before": true,
+  "dropout": 0.1,
+  "enable_bias": true,
+  "eos_token_id": 2,
+  "ffn_dim": 8192,
+  "hidden_size": 2048,
+  "init_std": 0.02,
+  "layer_norm_elementwise_affine": true,
+  "layerdrop": 0.0,
+  "learned_embeddings": true,
+  "max_position_embeddings": 2048,
+  "model_type": "opt",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 24,
+  "pad_token_id": 1,
+  "scale_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.25.1",
+  "use_cache": true,
+  "vocab_size": 50000,
+  "word_embed_proj_dim": 2048
+}

flax_model.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:055d597d7985d15a748fda64122e254f01103ee8c99562c79d2fc23352a8b589
+size 2630415559

special_tokens_map.json

@@ -0,0 +1,5 @@
+{
+  "bos_token": "</s>",
+  "eos_token": "</s>",
+  "pad_token": "<pad>"
+}

tokenizer_config.json

@@ -0,0 +1,6 @@
+{
+  "model_max_length": 1000000000000000019884624838656,
+  "name_or_path": "facebook/galactica-1.3b",
+  "special_tokens_map_file": "/content/tokenizer/special_tokens_map.json",
+  "tokenizer_class": "PreTrainedTokenizerFast"
+}