Add files using upload-large-folder tool

README.md

@@ -1,7 +1,8 @@
 ---
 library_name: diffusers
 tags:
-- src-ai
+- pruna_pro-ai
+- safetensors
 ---
 
 # Model Card for PrunaAI/test-save-tiny-stable-diffusion-pipe-smashed-pro
@@ -13,7 +14,7 @@ This model was created using the [pruna](https://github.com/PrunaAI/pruna) libra
 First things first, you need to install the pruna library:
 
 ```bash
-pip install src
+pip install pruna_pro
 ```
 
 You can [use the diffusers library to load the model](https://huggingface.co/PrunaAI/test-save-tiny-stable-diffusion-pipe-smashed-pro?library=diffusers) but this might not include all optimizations by default.
@@ -21,14 +22,17 @@ You can [use the diffusers library to load the model](https://huggingface.co/Pru
 To ensure that all optimizations are applied, use the pruna library to load the model using the following code:
 
 ```python
-from src import PrunaProModel
+from pruna_pro import PrunaProModel
 
-loaded_model = PrunaProModel.from_hub(
+loaded_model = PrunaProModel.from_pretrained(
     "PrunaAI/test-save-tiny-stable-diffusion-pipe-smashed-pro"
 )
+# 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
 
@@ -43,6 +47,7 @@ The compression configuration of the model is stored in the `smash_config.json`
     "distributer": null,
     "enhancer": null,
     "factorizer": null,
+    "kernel": null,
     "pruner": null,
     "quantizer": null,
     "recoverer": null,
@@ -58,6 +63,7 @@ The compression configuration of the model is stored in the `smash_config.json`
         "pruner": null,
         "quantizer": null,
         "distiller": null,
+        "kernel": null,
         "cacher": null,
         "recoverer": null,
         "distributer": null,
@@ -74,4 +80,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

@@ -1,4 +1,5 @@
 {
+  "_name_or_path": "/Users/davidberenstein/.cache/huggingface/hub/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/3ee6c9f225f088ad5d35b624b6514b091e6a4849/safety_checker",
   "architectures": [
     "StableDiffusionSafetyChecker"
   ],
@@ -12,40 +13,28 @@
     "dropout": 0.1,
     "eos_token_id": 2,
     "gradient_checkpointing": false,
-    "hidden_act": "quick_gelu",
     "hidden_size": 32,
-    "initializer_factor": 1.0,
-    "initializer_range": 0.02,
     "intermediate_size": 37,
-    "layer_norm_eps": 1e-05,
     "max_position_embeddings": 512,
     "model_type": "clip_text_model",
     "num_attention_heads": 4,
     "num_hidden_layers": 5,
-    "projection_dim": 512,
-    "torch_dtype": "float32",
     "vocab_size": 99
   },
   "torch_dtype": "float32",
-  "transformers_version": "4.53.3",
+  "transformers_version": "4.48.0",
   "vision_config": {
+    "_attn_implementation_autoset": true,
     "attention_dropout": 0.1,
     "dropout": 0.1,
     "gradient_checkpointing": false,
-    "hidden_act": "quick_gelu",
     "hidden_size": 32,
     "image_size": 30,
-    "initializer_factor": 1.0,
-    "initializer_range": 0.02,
     "intermediate_size": 37,
-    "layer_norm_eps": 1e-05,
     "model_type": "clip_vision_model",
     "num_attention_heads": 4,
-    "num_channels": 3,
     "num_hidden_layers": 5,
-    "patch_size": 2,
-    "projection_dim": 512,
-    "torch_dtype": "float32"
+    "patch_size": 2
   },
   "vocab_size": 1000
 }

smash_config.json

@@ -6,6 +6,7 @@
     "distributer": null,
     "enhancer": null,
     "factorizer": null,
+    "kernel": null,
     "pruner": null,
     "quantizer": null,
     "recoverer": null,
@@ -21,6 +22,7 @@
         "pruner": null,
         "quantizer": null,
         "distiller": null,
+        "kernel": null,
         "cacher": null,
         "recoverer": null,
         "distributer": null,

text_encoder/config.json

@@ -1,4 +1,5 @@
 {
+  "_name_or_path": "/Users/davidberenstein/.cache/huggingface/hub/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/3ee6c9f225f088ad5d35b624b6514b091e6a4849/text_encoder",
   "architectures": [
     "CLIPTextModel"
   ],
@@ -19,6 +20,6 @@
   "pad_token_id": 1,
   "projection_dim": 512,
   "torch_dtype": "float32",
-  "transformers_version": "4.53.3",
+  "transformers_version": "4.48.0",
   "vocab_size": 1000
 }

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.34.0",
-  "_name_or_path": "/home/runner/.cache/huggingface/hub/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/3ee6c9f225f088ad5d35b624b6514b091e6a4849/unet",
+  "_name_or_path": "/Users/davidberenstein/.cache/huggingface/hub/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/3ee6c9f225f088ad5d35b624b6514b091e6a4849/unet",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.34.0",
-  "_name_or_path": "/home/runner/.cache/huggingface/hub/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/3ee6c9f225f088ad5d35b624b6514b091e6a4849/vae",
+  "_name_or_path": "/Users/davidberenstein/.cache/huggingface/hub/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/3ee6c9f225f088ad5d35b624b6514b091e6a4849/vae",
   "act_fn": "silu",
   "block_out_channels": [
     32,