Vittorio Pippi
commited on
Commit
·
482b875
1
Parent(s):
a2dae75
Initial commit
Browse files- modeling_emuru.py +1 -1
modeling_emuru.py
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
import torch
|
3 |
import torch.nn as nn
|
4 |
from transformers import PreTrainedModel, T5ForConditionalGeneration, T5Config, AutoTokenizer
|
5 |
-
from configuration_emuru import EmuruConfig
|
6 |
from diffusers import AutoencoderKL
|
7 |
from einops.layers.torch import Rearrange
|
8 |
from einops import rearrange, repeat
|
|
|
2 |
import torch
|
3 |
import torch.nn as nn
|
4 |
from transformers import PreTrainedModel, T5ForConditionalGeneration, T5Config, AutoTokenizer
|
5 |
+
from .configuration_emuru import EmuruConfig
|
6 |
from diffusers import AutoencoderKL
|
7 |
from einops.layers.torch import Rearrange
|
8 |
from einops import rearrange, repeat
|