snicolau commited on
Commit
f11eb11
Β·
verified Β·
1 Parent(s): e7cd3b9

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

Browse files
lama/saicinpainting/training/trainers/__init__.py CHANGED
@@ -1,6 +1,6 @@
1
  import logging
2
  import torch
3
- from saicinpainting.training.trainers.default import DefaultInpaintingTrainingModule
4
 
5
 
6
  def get_training_model_class(kind):
 
1
  import logging
2
  import torch
3
+ from lama.saicinpainting.training.trainers.default import DefaultInpaintingTrainingModule
4
 
5
 
6
  def get_training_model_class(kind):