Add safetensors SD 1.5 tensor names mapping

google-ai-edge / ai-edge-torch

Supporting PyTorch models with the Google AI Edge TFLite runtime.

Apache License 2.0

278 stars 36 forks source link

Closed yichunk closed 1 month ago

yichunk commented 2 months ago

Add tensor names mapping from original SD 1.5 huggingface repo https://huggingface.co/runwayml/stable-diffusion-v1-5
Fix some miscellaneous typos

BUG=b/350994822

yichunk commented 1 month ago

just to confirm, have you tested by running a few iterations and check the generate image?

yes, verify it works with generated image.