From 9c788944cdb398388c6370217187a2f53dfefa7d Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Sun, 27 Aug 2023 14:30:58 +0900 Subject: [PATCH] upgrade olive-ai to 0.3.1. --- configs/olive_optimize/config_safety_checker.json | 1 + configs/olive_optimize/config_text_encoder.json | 1 + configs/olive_optimize/config_unet.json | 1 + configs/olive_optimize/config_vae_decoder.json | 1 + configs/olive_optimize/config_vae_encoder.json | 1 + requirements_olive.txt | 13 ++++++++++++- 6 files changed, 17 insertions(+), 1 deletion(-) diff --git a/configs/olive_optimize/config_safety_checker.json b/configs/olive_optimize/config_safety_checker.json index b9dffc2cef8..ac7a0a36322 100644 --- 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 index 53bdae4d5cf..7d4f976376f 100644 --- 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 index 5a78be26853..42170c60efe 100644 --- 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 index fe3a9fccbb1..283360866ee 100644 --- 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 index 0f3510b8171..6a7bf681858 100644 --- 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 index e36bf65f605..79b4fc7ea19 100644 --- 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 \ No newline at end of file +torchmetrics~=0.10.0 +transformers \ No newline at end of file