-- This commit adds `batch_size` command-line arg.
-- It also involves replicating the prompt `batch_size` no. of times.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
-- Currently we require users to specify the base model on which the custom
model (.ckpt) is tuned on. Even for running a HuggingFace repo-id, we
require the users to go a tedious way of adding things to variants.json.
-- This commit aims to address the above issues and will be treated as a
starting point for a series of design changes which makes using SHARK's SD
easier.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
-- This commit fixes CKPT script to rely on the previous CKPT to Diffusers
script.
TODO: Let go of the script once the CKPT is included in next release
of diffusers.
-- It also adds many variants as part of `variants.json` and updates
`README.md` to reflect change in default `hf_model_id`.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
-- This commit makes using custom models easier using a combination of
`import_mlir`, `ckpt_loc` and `hf_model_id`.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
-- This commit introduces a fix for .vmfb naming to strip away any
non-alphanumeric characters from `custom_model` path.
-- It also updates the README.md to include the `custom_model` arg.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
* Increase JPEG output quality & disable subsampling
Increased to JPEG95 from the default JPEG75 which is way too compressed. Output image size is now ~100kb. Previously was ~20kb.
* Increase JPEG output quality & disable subsampling
Add jpeg quality increase on cli
* line length changes
* line length changes
* Add script to auto annotate SD models and variants
* Add model config files
* Add script to auto annotate SD models and variants
* Add model config files
* Move config files to shark_tank
* [SharkInference] Make SharkInference compile the entire module
-- Previously SharkInference was compiling and providing run APIs
for a harcoded function with function name "forward".
-- This commit makes the compiling functionality generic and now
any function being defined within the module can be run.
-- It also creates an API to fetch all the function names defined
within the compiled module.
-- This commit updates both web and command-line execution of Stable
Diffusion to use new API of SharkInference.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
