Не видит высоту картинки из Firestore в CollectionView (PinterestLayout)

Есть проект с CollectionView(PinterestLayout), то есть высота cell зависит от высоты картинки не обрезая при этом ее по ширине, как тут: https://ibb.co/J3KdDtG, я пытаюсь сделать тоже самое в проекте где картинка берется из Firestore (то есть, это ссылка), картинку он видит, но с размерами беда как тут: https://ibb.co/JzXZv3V, знаю что проблема скорее всего в этой части кода, не пойму что нужно делать, спасибо.

       {
           let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "CustomCell", for: indexPath) as! CustomCell
           cell.imageView.sd_setImage(with: URL(string: flovers[indexPath.row].floverImg!))
           cell.label.text = flovers[indexPath.row].floverName
         //  cell.descriptionLabel.text = flovers[indexPath.row].floverName
           cell.layer.cornerRadius = 15

          
           return cell
       }```

тут все ок, картинку он берет
а тут мне кажется проблема

```{
   func collectionView(collectionView: UICollectionView, heightForPhotoAt indexPath: IndexPath, with width: CGFloat) -> CGFloat {
       
       if let post = posts?[indexPath.item], let photo = post.floverImg {
           let boundingRect = CGRect(x: 0, y: 0, width: width, height: CGFloat(MAXFLOAT))
           let rect = AVMakeRect(aspectRatio: photo.size(), insideRect: boundingRect)
          
           return CGFloat(rect.size.height)
       }
       
       return 0
   }```

return CGFloat(rect.size.height) не работает, сразу идет к return 0(ну и соответственно ничего не показывает, показывает когда меняю значение)