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):
|