Vittorio Pippi commited on
Commit
2226641
·
1 Parent(s): 2fa529a

Update .gitignore to include __pycache__; set config_class in Emuru model

Browse files
.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
  """