Skip to content

Commit e3e4ea5

Browse files
Update README.md
-- Make usage of `hf_model_id` clearer.
1 parent a3e4ea3 commit e3e4ea5

File tree

1 file changed

+2
-2
lines changed
  • shark/examples/shark_inference/stable_diffusion

1 file changed

+2
-2
lines changed

shark/examples/shark_inference/stable_diffusion/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,9 @@ pip install omegaconf safetensors pytorch_lightning
2424
```
2525
* Download a [.ckpt](https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.0-pruned-fp32.ckpt) file in case you don't have a locally generated `.ckpt` file for StableDiffusion.
2626

27-
* Now pass the above `.ckpt` file to `ckpt_loc` command-line argument using the following :-
27+
* Now pass the above `.ckpt` file to `ckpt_loc` command-line argument using the following (note the `hf_model_id` flag which states what the base model is from which the `.ckpt` model was fined-tuned off of) :-
2828
```shell
29-
python3.10 main.py --precision=fp16 --device=vulkan --prompt="tajmahal, oil on canvas, sunflowers, 4k, uhd" --max_length=64 --import_mlir --ckpt_loc="/path/to/.ckpt/file" --hf_model_id="<HuggingFace repo-id>"
29+
python3.10 main.py --precision=fp16 --device=vulkan --prompt="tajmahal, oil on canvas, sunflowers, 4k, uhd" --max_length=64 --import_mlir --ckpt_loc="/path/to/.ckpt/file" --hf_model_id="CompVis/stable-diffusion-v1-4"
3030
```
3131
* We use a combination of 3 flags to make this feature work : `import_mlir`, `ckpt_loc` and `hf_model_id`, of which `import_mlir` needs to be present. In case `ckpt_loc` is not specified then a [default](https://huggingface.co/stabilityai/stable-diffusion-2-1-base) HuggingFace repo-id is run via `hf_model_id`. So, you need to specify which base model's `.ckpt` you are using via `hf_model_id`.
3232

0 commit comments

Comments
 (0)