Add files using upload-large-folder tool

README.md

@@ -2,6 +2,7 @@
 library_name: diffusers
 tags:
 - pruna-ai
+- safetensors
 ---
 
 # Model Card for PrunaAI/test-save-tiny-stable-diffusion-pipe-smashed
@@ -23,12 +24,15 @@ To ensure that all optimizations are applied, use the pruna library to load the
 ```python
 from pruna import PrunaModel
 
-loaded_model = PrunaModel.from_hub(
+loaded_model = PrunaModel.from_pretrained(
     "PrunaAI/test-save-tiny-stable-diffusion-pipe-smashed"
 )
+# we can then run inference using the methods supported by the base model
 ```
 
-After loading the model, you can use the inference methods of the original model. Take a look at the [documentation](https://pruna.readthedocs.io/en/latest/index.html) for more usage information.
+
+For inference, you can use the inference methods of the original model like shown in [the original model card](https://huggingface.co/hf-internal-testing/tiny-stable-diffusion-pipe?library=diffusers).
+ Alternatively, you can visit [the Pruna documentation](https://docs.pruna.ai/en/stable/) for more information.
 
 ## Smash Configuration
 
@@ -40,10 +44,12 @@ The compression configuration of the model is stored in the `smash_config.json`
     "cacher": null,
     "compiler": null,
     "factorizer": null,
+    "kernel": null,
     "pruner": null,
     "quantizer": null,
     "batch_size": 1,
     "device": "cpu",
+    "device_map": null,
     "save_fns": [],
     "load_fns": [
         "diffusers"
@@ -52,6 +58,7 @@ The compression configuration of the model is stored in the `smash_config.json`
         "factorizer": null,
         "pruner": null,
         "quantizer": null,
+        "kernel": null,
         "cacher": null,
         "compiler": null,
         "batcher": null
@@ -65,4 +72,4 @@ The compression configuration of the model is stored in the `smash_config.json`
 [![GitHub](https://img.shields.io/github/followers/PrunaAI?label=Follow%20%40PrunaAI&style=social)](https://github.com/PrunaAI)
 [![LinkedIn](https://img.shields.io/badge/LinkedIn-Connect-blue)](https://www.linkedin.com/company/93832878/admin/feed/posts/?feedType=following)
 [![Discord](https://img.shields.io/badge/Discord-Join%20Us-blue?style=social&logo=discord)](https://discord.com/invite/rskEr4BZJx)
-[![Reddit](https://img.shields.io/reddit/subreddit-subscribers/PrunaAI?style=social)](https://www.reddit.com/r/PrunaAI/)
+[![Reddit](https://img.shields.io/reddit/subreddit-subscribers/PrunaAI?style=social)](https://www.reddit.com/r/PrunaAI/)

safety_checker/config.json

@@ -27,7 +27,7 @@
     "vocab_size": 99
   },
   "torch_dtype": "float32",
-  "transformers_version": "4.52.4",
+  "transformers_version": "4.52.3",
   "vision_config": {
     "attention_dropout": 0.1,
     "dropout": 0.1,

smash_config.json

@@ -3,10 +3,12 @@
     "cacher": null,
     "compiler": null,
     "factorizer": null,
+    "kernel": null,
     "pruner": null,
     "quantizer": null,
     "batch_size": 1,
     "device": "cpu",
+    "device_map": null,
     "save_fns": [],
     "load_fns": [
         "diffusers"
@@ -15,6 +17,7 @@
         "factorizer": null,
         "pruner": null,
         "quantizer": null,
+        "kernel": null,
         "cacher": null,
         "compiler": null,
         "batcher": null

text_encoder/config.json

@@ -19,6 +19,6 @@
   "pad_token_id": 1,
   "projection_dim": 512,
   "torch_dtype": "float32",
-  "transformers_version": "4.52.4",
+  "transformers_version": "4.52.3",
   "vocab_size": 1000
 }