upgrade olive-ai to 0.3.1.

lshqqytiger · Aug 27, 2023 · 9c78894 · 9c78894
1 parent 08994e3
commit 9c78894
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 1 deletion.
diff --git a/configs/olive_optimize/config_safety_checker.json b/configs/olive_optimize/config_safety_checker.json
@@ -59,6 +59,7 @@
     },
     "optimize": {
       "type": "OrtTransformersOptimization",
+      "disable_search": true,
       "config": {
         "model_type": "unet",
         "float16": true,

diff --git a/configs/olive_optimize/config_text_encoder.json b/configs/olive_optimize/config_text_encoder.json
@@ -46,6 +46,7 @@
     },
     "optimize": {
       "type": "OrtTransformersOptimization",
+      "disable_search": true,
       "config": {
         "model_type": "clip",
         "float16": true,

diff --git a/configs/olive_optimize/config_unet.json b/configs/olive_optimize/config_unet.json
@@ -66,6 +66,7 @@
     },
     "optimize": {
       "type": "OrtTransformersOptimization",
+      "disable_search": true,
       "config": {
         "model_type": "unet",
         "float16": true,

diff --git a/configs/olive_optimize/config_vae_decoder.json b/configs/olive_optimize/config_vae_decoder.json
@@ -53,6 +53,7 @@
     },
     "optimize": {
       "type": "OrtTransformersOptimization",
+      "disable_search": true,
       "config": {
         "model_type": "vae",
         "float16": true,

diff --git a/configs/olive_optimize/config_vae_encoder.json b/configs/olive_optimize/config_vae_encoder.json
@@ -53,6 +53,7 @@
     },
     "optimize": {
       "type": "OrtTransformersOptimization",
+      "disable_search": true,
       "config": {
         "model_type": "vae",
         "float16": true,

diff --git a/requirements_olive.txt b/requirements_olive.txt
@@ -1,4 +1,15 @@
+accelerate
+diffusers
+numpy
+onnx
+optuna
+optimum
+pandas
+pydantic<2.0.0
+protobuf<4.0.0
+pyyaml
 torch==1.13.1
 torchvision==0.14.1
 torch-directml==0.1.13.1.dev230413
-protobuf==3.20.3
+torchmetrics~=0.10.0
+transformers