snicolau commited on
Commit
76be17b
Β·
verified Β·
1 Parent(s): 141cd5b

Update lama/saicinpainting/training/modules/__init__.py

Browse files
lama/saicinpainting/training/modules/__init__.py CHANGED
@@ -1,7 +1,7 @@
1
  import logging
2
 
3
- from saicinpainting.training.modules.ffc import FFCResNetGenerator
4
- from saicinpainting.training.modules.pix2pixhd import GlobalGenerator, MultiDilatedGlobalGenerator, \
5
  NLayerDiscriminator, MultidilatedNLayerDiscriminator
6
 
7
  def make_generator(config, kind, **kwargs):
 
1
  import logging
2
 
3
+ from lama.saicinpainting.training.modules.ffc import FFCResNetGenerator
4
+ from lama.saicinpainting.training.modules.pix2pixhd import GlobalGenerator, MultiDilatedGlobalGenerator, \
5
  NLayerDiscriminator, MultidilatedNLayerDiscriminator
6
 
7
  def make_generator(config, kind, **kwargs):