Upload folder using huggingface_hub

README.md

@@ -0,0 +1,63 @@
+---
+library_name: transformers
+pipeline_tag: text-generation
+inference: true
+widget:
+- text: Hello!
+  example_title: Hello world
+  group: Python
+---
+
+This model is for debugging. It is randomly initialized using the config from [mistralai/mathstral-7B-v0.1](https://huggingface.co/mistralai/mathstral-7B-v0.1) but with smaller size. 
+
+Codes:
+```python
+from huggingface_hub import create_repo, upload_folder
+from transformers import (
+    pipeline,
+    set_seed,
+    AutoConfig,
+    AutoModelForCausalLM,
+    AutoTokenizer,
+    GenerationConfig,
+)
+import torch
+import transformers
+import os
+
+model_id = "mistralai/mathstral-7B-v0.1"
+repo_id = "yujiepan/mathstral-v0.1-tiny-random"
+save_path = f"/tmp/{repo_id}"
+
+config = AutoConfig.from_pretrained(model_id)
+config.hidden_size = 8
+config.intermediate_size = 32
+config.num_attention_heads = 4
+config.num_hidden_layers = 2
+config.num_key_value_heads = 2
+config.head_dim = 2
+print(config)
+
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+tokenizer.save_pretrained(save_path)
+
+model = AutoModelForCausalLM.from_config(config, torch_dtype=torch.bfloat16)
+model.generation_config = GenerationConfig.from_pretrained(model_id)
+
+set_seed(42)
+with torch.no_grad():
+    for _, p in sorted(model.named_parameters()):
+        torch.nn.init.uniform_(p, -0.1, 0.1)
+
+model.save_pretrained(save_path)
+
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, do_sample=False, device="cuda")
+print(pipe("Hello World!"))
+
+messages = [
+    {"role": "system", "content": "You are a robot."},
+    {"role": "user", "content": "Hi!"},
+]
+chatbot = pipeline("text-generation", model=save_path, max_length=1000, max_new_tokens=16)
+print(chatbot(messages))
+```

config.json

@@ -0,0 +1,27 @@
+{
+  "_name_or_path": "mistralai/mathstral-7B-v0.1",
+  "architectures": [
+    "MistralForCausalLM"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "head_dim": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8,
+  "initializer_range": 0.02,
+  "intermediate_size": 32,
+  "max_position_embeddings": 32768,
+  "model_type": "mistral",
+  "num_attention_heads": 4,
+  "num_hidden_layers": 2,
+  "num_key_value_heads": 2,
+  "rms_norm_eps": 1e-05,
+  "rope_theta": 1000000.0,
+  "sliding_window": null,
+  "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.43.0.dev0",
+  "use_cache": true,
+  "vocab_size": 32768
+}

generation_config.json

@@ -0,0 +1,6 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "transformers_version": "4.43.0.dev0"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:889ed1d5d4ad3c6a1a204b49dcf13c9980147a1c31fdff08ebd455171e4b41d1
+size 1054672

special_tokens_map.json

@@ -0,0 +1,23 @@
+{
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer.model

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:59f95e28944c062244741268596badc900df86c7f5ded05088d2da22a7379e06
+size 587583