apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(docs): note about checking LoRA metadata keys

This commit is contained in:
Sean Sube 2023-02-19 12:29:15 -06:00
parent 0a72ec1567
commit 7ef63e14c4
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
docs/converting-models.md
View File

@ -116,7 +116,19 @@ which can be converted into a diffusers directory and finally ONNX models.
Weights exported by the two repositories are not compatible with the other and you must use the same scripts that Weights exported by the two repositories are not compatible with the other and you must use the same scripts that
originally created a set of weights to merge them. originally created a set of weights to merge them.
Try the other repository if you get an error about missing metadata, for example: If you have a `.safetensors` file, check the metadata keys:
```python
>>> import safetensors
>>> t = safetensors.safe_open("/home/ssube/lora-weights/jack.safetensors", framework="pt")
>>> print(t.metadata())
{'ss_batch_size_per_device': '1', 'ss_bucket_info': 'null', 'ss_cache_latents': 'True', 'ss_clip_skip': '2', ...}
```
If they start with `lora_`, it's probably from the `cloneofsimo/lora` scripts. If they start with `ss_`, it's
probably from the `kohya-ss/sd-scripts` scripts.
If you get an error about missing metadata, try the other repository. For example:
```none ```none
warnings.warn( warnings.warn(