Vittorio Pippi
commited on
Commit
·
2226641
1
Parent(s):
2fa529a
Update .gitignore to include __pycache__; set config_class in Emuru model
Browse files- .gitignore +2 -1
- __pycache__/configuration_emuru.cpython-311.pyc +0 -0
- modeling_emuru.py +1 -1
.gitignore
CHANGED
@@ -3,4 +3,5 @@ test.py
|
|
3 |
model.py
|
4 |
visual_prompting.py
|
5 |
test.png
|
6 |
-
output.png
|
|
|
|
3 |
model.py
|
4 |
visual_prompting.py
|
5 |
test.png
|
6 |
+
output.png
|
7 |
+
__pycache__
|
__pycache__/configuration_emuru.cpython-311.pyc
CHANGED
Binary files a/__pycache__/configuration_emuru.cpython-311.pyc and b/__pycache__/configuration_emuru.cpython-311.pyc differ
|
|
modeling_emuru.py
CHANGED
@@ -10,6 +10,7 @@ from typing import Optional, Tuple, List, Any
|
|
10 |
from PIL import Image
|
11 |
|
12 |
class Emuru(PreTrainedModel):
|
|
|
13 |
"""
|
14 |
Emuru is a conditional generative model that integrates a T5-based decoder with a VAE
|
15 |
for image generation conditioned on text and style images.
|
@@ -28,7 +29,6 @@ class Emuru(PreTrainedModel):
|
|
28 |
z_rearrange (Rearrange): Layer to rearrange T5 outputs back to VAE latent dimensions.
|
29 |
mse_criterion (nn.MSELoss): Mean squared error loss function.
|
30 |
"""
|
31 |
-
config_class = EmuruConfig
|
32 |
|
33 |
def __init__(self, config: EmuruConfig) -> None:
|
34 |
"""
|
|
|
10 |
from PIL import Image
|
11 |
|
12 |
class Emuru(PreTrainedModel):
|
13 |
+
config_class = EmuruConfig
|
14 |
"""
|
15 |
Emuru is a conditional generative model that integrates a T5-based decoder with a VAE
|
16 |
for image generation conditioned on text and style images.
|
|
|
29 |
z_rearrange (Rearrange): Layer to rearrange T5 outputs back to VAE latent dimensions.
|
30 |
mse_criterion (nn.MSELoss): Mean squared error loss function.
|
31 |
"""
|
|
|
32 |
|
33 |
def __init__(self, config: EmuruConfig) -> None:
|
34 |
"""
|