Nupur Kumari commited on
Commit
bc96fce
1 Parent(s): d2113d1

custom-diffusion-space

Browse files
Files changed (1) hide show
  1. inference.py +3 -1
inference.py CHANGED
@@ -11,7 +11,9 @@ import numpy as np
11
  import torch
12
  from diffusers import StableDiffusionPipeline
13
  sys.path.insert(0, './custom-diffusion')
14
- from custom-diffusion.src import diffuser_training
 
 
15
 
16
 
17
  class InferencePipeline:
 
11
  import torch
12
  from diffusers import StableDiffusionPipeline
13
  sys.path.insert(0, './custom-diffusion')
14
+ sys.path.insert(0, './custom-diffusion/src')
15
+
16
+ import diffuser_training
17
 
18
 
19
  class InferencePipeline: