nekoshadow commited on
Commit
16bf801
·
1 Parent(s): 4f2a492
ldm/models/diffusion/sync_dreamer.py CHANGED
@@ -9,12 +9,12 @@ from skimage.io import imsave
9
  from torch.optim.lr_scheduler import LambdaLR
10
  from tqdm import tqdm
11
 
12
- from SyncDreamer.ldm.base_utils import read_pickle, concat_images_list
13
- from SyncDreamer.ldm.models.diffusion.sync_dreamer_utils import get_warp_coordinates, create_target_volume
14
- from SyncDreamer.ldm.models.diffusion.sync_dreamer_network import NoisyTargetViewEncoder, SpatialTime3DNet, FrustumTV3DNet
15
- from SyncDreamer.ldm.modules.diffusionmodules.util import make_ddim_timesteps, timestep_embedding
16
- from SyncDreamer.ldm.modules.encoders.modules import FrozenCLIPImageEmbedder
17
- from SyncDreamer.ldm.util import instantiate_from_config
18
 
19
  def disabled_train(self, mode=True):
20
  """Overwrite model.train with this function to make sure train/eval mode
 
9
  from torch.optim.lr_scheduler import LambdaLR
10
  from tqdm import tqdm
11
 
12
+ from ...base_utils import read_pickle, concat_images_list
13
+ from .sync_dreamer_utils import get_warp_coordinates, create_target_volume
14
+ from .sync_dreamer_network import NoisyTargetViewEncoder, SpatialTime3DNet, FrustumTV3DNet
15
+ from ...modules.diffusionmodules.util import make_ddim_timesteps, timestep_embedding
16
+ from ...modules.encoders.modules import FrozenCLIPImageEmbedder
17
+ from ...util import instantiate_from_config
18
 
19
  def disabled_train(self, mode=True):
20
  """Overwrite model.train with this function to make sure train/eval mode