ysharma HF staff commited on
Commit
224bd87
1 Parent(s): 6464abf
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -5,8 +5,8 @@ import torch.nn as nn
5
  import numpy as np
6
  from einops import rearrange
7
 
8
- from ldm.util import instantiate_from_config
9
- from ldm.modules.attention import LinearAttention
10
 
11
 
12
  def get_timestep_embedding(timesteps, embedding_dim):
5
  import numpy as np
6
  from einops import rearrange
7
 
8
+ from util import instantiate_from_config
9
+ from attention import LinearAttention
10
 
11
 
12
  def get_timestep_embedding(timesteps, embedding_dim):