spotted a possible minor type annotation error

[imchkkim]

hi thx for great work.

I think I spotted a possible minor type annotation error in

ziplora_pytorch/utils.py line 14 where returns

)-> torch.Tensor

get_lora_weights function seems to returns tensors dict ( tensors[key] = f.get_tensor(key) in line 38 )

therefore, I assume the return type of function should be

) -> Dict[str, torch.Tensor]:

also in line 116

tensors: torch.Tensor, key: str, prefix: str = "unet.unet."
# torch.Tensor -> Dict[str, torch.Tensor]

[mkshing]

Thanks! Fixed.

• https://github.com/mkshing/ziplora-pytorch/commit/abe2406eb048307f5dc4b8047a8418f216691134
• https://github.com/mkshing/ziplora-pytorch/commit/0e39ceb7ee1cde3c98a57e690eadebc638a52994