Upload folder using huggingface_hub (#6)

- 73bfe71853d28a1623c176152ac2b90c2147df66397a912d789ed54596d8333e (45f7ddc5d9f11703690a4d9e5190f2b116a0bd13)
- 571b8d8c3a16f07de64314222cf87cac8c334f1f124c31d90213e8ce00cf7450 (28eccc0771a5544eefc0f84a9cbd8943310d534a)

README.md

@@ -37,16 +37,17 @@ metrics:
 ![image info](./plots.png)
 
 **Important remarks:**
-- The quality of the model output might slightly vary compared to the base model. There might be minimal quality loss.
+- The quality of the model output might slightly vary compared to the base model.
 - These results were obtained on NVIDIA A100-PCIE-40GB with configuration described in config.json and are obtained after a hardware warmup. Efficiency results may vary in other settings (e.g. other hardware, image size, batch size, ...).
 - You can request premium access to more compression methods and tech support for your specific use-cases [here](https://z0halsaff74.typeform.com/pruna-access?typeform-source=www.pruna.ai).
+- Results mentioning "first" are obtained after the first run of the model. The first run might take more memory or be slower than the subsequent runs due cuda overheads.
 
 ## Setup
 
 You can run the smashed model with these steps:
 
-0. Check cuda, torch, packaging requirements are installed. For cuda, check with `nvcc --version` and install with `conda install nvidia/label/cuda-12.1.0::cuda`. For packaging and torch, run `pip install packaging torch`.
-1. Install the `pruna-engine` available [here](https://pypi.org/project/pruna-engine/) on Pypi. It might take 15 minutes to install.
+0. Check that you have linux, python 3.10, and cuda 12.1.0 requirements installed. For cuda, check with `nvcc --version` and install with `conda install nvidia/label/cuda-12.1.0::cuda`.
+1. Install the `pruna-engine` available [here](https://pypi.org/project/pruna-engine/) on Pypi. It might take up to 15 minutes to install.
     ```bash
    pip install pruna-engine[gpu]==0.6.0 --extra-index-url https://pypi.nvidia.com --extra-index-url https://pypi.ngc.nvidia.com --extra-index-url https://prunaai.pythonanywhere.com/
     ```

model/optimized_model.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:24b558d960b9f48ee3094734a6552713b5f25dae6402f04c718a1c1b60929a6e
-size 2743425953
+oid sha256:ac5cbf06b1783c85051877a4916e957ac6cc5158b8681567c5db77f6ce9a4e28
+size 2743426080