mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-11-17 09:25:47 +00:00
a338902ab5
Loading png images for captioning will cause the following error since PIL loads png images in RGBA mode.
```python
File "./miniconda3/lib/python3.8/site-packages/torchvision/transforms/functional_tensor.py", line 940, in normalize
return tensor.sub_(mean).div_(std)
RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0
```
|
||
|---|---|---|
| .. | ||
| phraselists | ||
| __init__.py | ||
| blur_detect.py | ||
| bool_masker.py | ||
| clip_masking.py | ||
| describe_image_blip.py | ||
| describe_image_clip.py | ||
| face_restoration_codeformer.py | ||
| facecrop.py | ||
| prompt_expansion.py | ||
| upscale_realesrgan.py | ||