search

google-ai-edge / ai-edge-torch

Supporting PyTorch models with the Google AI Edge TFLite runtime.
Apache License 2.0
199 stars 23 forks source link

Fix broken decoder conversion of SD. #51

Closed yichunk closed 3 weeks ago

yichunk commented 3 weeks ago
  1. Fix UpDecoder config.
  2. Remove ModuleList of optional blocks. nn.Module seems cannot contain empty ModuleList

BUG=b/311216181