Spaces:
Runtime error
Runtime error
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
|
9 |
-
from
|
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):
|