ml-stable-diffusion
ml-stable-diffusion copied to clipboard
Published
20 hours ago •
apple
apple
FileNotFoundError: text_encoder CoreML model doesn't exist at models/coreml-stable-diffusion-v1-4_original_packages/Stable_Diffusion_version_CompVis_stable-diffusion-v1-4_text_encoder.mlpackage
1. download.py:
from huggingface_hub import snapshot_download
from pathlib import Path
repo_id = "apple/coreml-stable-diffusion-v1-4"
variant = "original/packages"
model_path = Path("./models") / (repo_id.split("/")[-1] + "_" + variant.replace("/", "_"))
snapshot_download(repo_id, allow_patterns=f"{variant}/*", local_dir=model_path, local_dir_use_symlinks=False)
print(f"Model downloaded at {model_path}")
2. download model:
(apple-sd-env) jimmy@Jimmys-Air ml-stable-diffusion % python download.py
Downloading (…)ackage/Manifest.json: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████| 617/617 [00:00<00:00, 571kB/s]
Downloading (…)ackage/Manifest.json: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████| 617/617 [00:00<00:00, 531kB/s]
Downloading (…)ackage/Manifest.json: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████| 617/617 [00:00<00:00, 214kB/s]
Downloading model.mlmodel: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████| 146k/146k [00:00<00:00, 543kB/s]
Downloading model.mlmodel: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████| 289k/289k [00:00<00:00, 764kB/s]
Downloading model.mlmodel: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████| 147k/147k [00:00<00:00, 518kB/s]
Downloading (…)ackage/Manifest.json: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████| 617/617 [00:00<00:00, 178kB/s]
Downloading model.mlmodel: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████| 723k/723k [00:01<00:00, 653kB/s]
Downloading (…)ackage/Manifest.json: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████| 617/617 [00:00<00:00, 194kB/s]
Downloading model.mlmodel: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████| 117k/117k [00:00<00:00, 276kB/s]
Downloading weight.bin: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████| 68.4M/68.4M [00:04<00:00, 16.6MB/s]
Downloading weight.bin: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████| 99.0M/99.0M [00:10<00:00, 9.31MB/s]
Downloading weight.bin: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████| 246M/246M [00:20<00:00, 11.9MB/s]
Downloading weight.bin: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████| 608M/608M [00:30<00:00, 19.9MB/s]
Downloading weight.bin: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████| 1.72G/1.72G [00:59<00:00, 29.1MB/s]
Fetching 15 files: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 15/15 [01:00<00:00, 4.06s/it]
Model downloaded at models/coreml-stable-diffusion-v1-4_original_packages████▉ | 786M/1.72G [00:30<00:32, 28.7MB/s]
3. Run
(apple-sd-env) jimmy@Jimmys-Air ml-stable-diffusion % python -m python_coreml_stable_diffusion.pipeline --prompt "a photo of an astronaut riding a horse on mars" -i models/coreml-stable-diffusion-v1-4_original_packages -o images/ --compute-unit ALL --seed 93
INFO:__main__:Setting random seed to 93
INFO:__main__:Initializing PyTorch pipe for reference configuration
/Users/jimmy/.pyenv/versions/apple-sd-env/lib/python3.11/site-packages/transformers/models/clip/feature_extraction_clip.py:28: FutureWarning: The class CLIPFeatureExtractor is deprecated and will be removed in version 5 of Transformers. Please use CLIPImageProcessor instead.
warnings.warn(
`text_config_dict` is provided which will be used to initialize `CLIPTextConfig`. The value `text_config["id2label"]` will be overriden.
INFO:__main__:Removed PyTorch pipe to reduce peak memory consumption
INFO:__main__:Loading Core ML models in memory from models/coreml-stable-diffusion-v1-4_original_packages
INFO:python_coreml_stable_diffusion.coreml_model:Loading text_encoder mlpackage
Traceback (most recent call last):
File "<frozen runpy>", line 198, in _run_module_as_main
File "<frozen runpy>", line 88, in _run_code
File "/Users/jimmy/dev/foxai/apple/ml-stable-diffusion/python_coreml_stable_diffusion/pipeline.py", line 651, in <module>
main(args)
File "/Users/jimmy/dev/foxai/apple/ml-stable-diffusion/python_coreml_stable_diffusion/pipeline.py", line 559, in main
coreml_pipe = get_coreml_pipe(pytorch_pipe=pytorch_pipe,
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/Users/jimmy/dev/foxai/apple/ml-stable-diffusion/python_coreml_stable_diffusion/pipeline.py", line 505, in get_coreml_pipe
coreml_pipe_kwargs.update({
^
File "/Users/jimmy/dev/foxai/apple/ml-stable-diffusion/python_coreml_stable_diffusion/pipeline.py", line 506, in <dictcomp>
model_name: _load_mlpackage(
^^^^^^^^^^^^^^^^
File "/Users/jimmy/dev/foxai/apple/ml-stable-diffusion/python_coreml_stable_diffusion/coreml_model.py", line 96, in _load_mlpackage
raise FileNotFoundError(
FileNotFoundError: text_encoder CoreML model doesn't exist at models/coreml-stable-diffusion-v1-4_original_packages/Stable_Diffusion_version_CompVis_stable-diffusion-v1-4_text_encoder.mlpackage
4. Error
FileNotFoundError: text_encoder CoreML model doesn't exist at models/coreml-stable-diffusion-v1-4_original_packages/Stable_Diffusion_version_CompVis_stable-diffusion-v1-4_text_encoder.mlpackage
