Upload folder using huggingface_hub (#3)

- c5dbc7b5d587a8c1a7133b8c2dd98205213173bc8288ee6a3514dfc1727128e9 (283d33db5ce5ff20924ba5b121d1cd92b97c5d93)

README.md

@@ -1,6 +1,6 @@
 ---
 thumbnail: "https://assets-global.website-files.com/646b351987a8d8ce158d1940/64ec9e96b4334c0e1ac41504_Logo%20with%20white%20text.svg"
-base_model: HuggingFaceTB/SmolLM2-1.7B-Instruct
+base_model: ORIGINAL_REPO_NAME
 metrics:
 - memory_disk
 - memory_inference
@@ -52,7 +52,7 @@ tags:
 
 You can run the smashed model with these steps:
 
-0. Check requirements from the original repo HuggingFaceTB/SmolLM2-1.7B-Instruct installed. In particular, check python, cuda, and transformers versions.
+0. Check requirements from the original repo ORIGINAL_REPO_NAME installed. In particular, check python, cuda, and transformers versions.
 1. Make sure that you have installed quantization related packages.
     ```bash
     pip install hqq
@@ -67,7 +67,7 @@ You can run the smashed model with these steps:
      model = HQQModelForCausalLM.from_quantized("PrunaAI/HuggingFaceTB-SmolLM2-1.7B-Instruct-HQQ-4bit-smashed", device_map='auto')
     except: 
      model = AutoHQQHFModel.from_quantized("PrunaAI/HuggingFaceTB-SmolLM2-1.7B-Instruct-HQQ-4bit-smashed")
-   tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM2-1.7B-Instruct")
+   tokenizer = AutoTokenizer.from_pretrained("ORIGINAL_REPO_NAME")
     
    input_ids = tokenizer("What is the color of prunes?,", return_tensors='pt').to(model.device)["input_ids"]
     
@@ -81,7 +81,7 @@ The configuration info are in `smash_config.json`.
 
 ## Credits & License
 
-The license of the smashed model follows the license of the original model. Please check the license of the original model HuggingFaceTB/SmolLM2-1.7B-Instruct before using this model which provided the base model. The license  of the `pruna-engine` is [here](https://pypi.org/project/pruna-engine/) on Pypi.
+The license of the smashed model follows the license of the original model. Please check the license of the original model ORIGINAL_REPO_NAME before using this model which provided the base model. The license  of the `pruna-engine` is [here](https://pypi.org/project/pruna-engine/) on Pypi.
 
 ## Want to compress other models?
 

config.json

@@ -1,6 +1,6 @@
 {
   "_attn_implementation_autoset": true,
-  "_name_or_path": "/home/ubuntu/.cache/pruna/tmpnc7f2rlo/tmpw0m8twle",
+  "_name_or_path": "HuggingFaceTB/SmolLM2-1.7B-Instruct",
   "architectures": [
     "LlamaForCausalLM"
   ],
@@ -21,38 +21,18 @@
   "num_key_value_heads": 32,
   "pad_token_id": 2,
   "pretraining_tp": 1,
-  "quantization_config": {
-    "quant_config": {
-      "offload_meta": false,
-      "scale_quant_params": null,
-      "weight_quant_params": {
-        "axis": 1,
-        "channel_wise": true,
-        "group_size": 64,
-        "nbits": 4,
-        "optimize": true,
-        "round_zero": true,
-        "view_as_float": false
-      },
-      "zero_quant_params": null
-    },
-    "quant_method": "hqq",
-    "skip_modules": [
-      "lm_head"
-    ]
-  },
   "rms_norm_eps": 1e-05,
   "rope_scaling": null,
   "rope_theta": 130000,
   "tie_word_embeddings": true,
-  "torch_dtype": "float16",
+  "torch_dtype": "bfloat16",
   "transformers.js_config": {
     "kv_cache_dtype": {
       "fp16": "float16",
       "q4f16": "float16"
     }
   },
-  "transformers_version": "4.46.3",
+  "transformers_version": "4.48.2",
   "use_cache": true,
   "vocab_size": 49152
 }

smash_config.json

@@ -1,34 +1,19 @@
 {
-    "comp_cgenerate_active": false,
-    "comp_ctranslate_active": false,
-    "comp_cwhisper_active": false,
-    "comp_diffusers2_active": false,
-    "comp_flux_caching_active": false,
-    "comp_ifw_active": false,
-    "comp_ipex_llm_active": false,
-    "comp_onediff_active": false,
-    "comp_step_caching_active": false,
-    "comp_torch_compile_active": false,
-    "comp_ws2t_active": false,
-    "comp_x-fast_active": false,
-    "prune_torch-structured_active": false,
-    "prune_torch-unstructured_active": false,
-    "quant_aqlm_active": false,
-    "quant_awq_active": false,
-    "quant_gptq_active": false,
-    "quant_half_active": false,
-    "quant_hqq_active": true,
-    "quant_llm-int8_active": false,
-    "quant_quanto_active": false,
-    "quant_torch_dynamic_active": false,
-    "quant_torch_static_active": false,
+    "batchers": null,
+    "cachers": null,
+    "compilers": null,
+    "distillers": null,
+    "pruners": null,
+    "quantizers": "hqq",
+    "recoverers": null,
     "quant_hqq_backend": "torchao_int4",
     "quant_hqq_group_size": 64,
     "quant_hqq_weight_bits": 4,
     "max_batch_size": 1,
     "device": "cuda",
-    "cache_dir": "/home/ubuntu/.cache/pruna/tmpnc7f2rlo",
+    "cache_dir": "/tmp/models/tmpimi9ply8",
     "task": "",
     "save_load_fn": "hqq",
-    "save_load_fn_args": {}
+    "save_load_fn_args": {},
+    "api_key": null
 }