snicolau commited on
Commit
67f2c43
β€’
1 Parent(s): 32a5e5a

Update lama/saicinpainting/training/losses/perceptual.py

Browse files
lama/saicinpainting/training/losses/perceptual.py CHANGED
@@ -3,7 +3,7 @@ import torch.nn as nn
3
  import torch.nn.functional as F
4
  import torchvision
5
 
6
- from models.ade20k import ModelBuilder
7
  from lama.saicinpainting.utils import check_and_warn_input_range
8
 
9
 
 
3
  import torch.nn.functional as F
4
  import torchvision
5
 
6
+ from lama.models.ade20k import ModelBuilder
7
  from lama.saicinpainting.utils import check_and_warn_input_range
8
 
9