Spaces:
Runtime error
Runtime error
File size: 1,218 Bytes
0c8d55e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
from transformers.configuration_utils import PretrainedConfig
from typing import Literal, Optional, Union
import json
class UnivaDenoiseTowerConfig(PretrainedConfig):
model_type = "univa_denoise_tower"
def __init__(
self,
denoiser_type: Literal["flux", "sd3"] = "flux",
denoise_projector_type: str = "mlp2x_gelu",
vae_projector_type: str = "mlp2x_gelu",
input_hidden_size: int = 1152,
vae_input_hidden_size: int = 64,
output_hidden_size: int = 4096,
denoiser_config: Optional[Union[str, dict]] = None,
**kwargs,
):
super().__init__(**kwargs)
self.denoiser_type = denoiser_type
self.denoise_projector_type = denoise_projector_type
self.vae_projector_type = vae_projector_type
self.input_hidden_size = input_hidden_size
self.vae_input_hidden_size = vae_input_hidden_size
self.output_hidden_size = output_hidden_size
self.denoiser_config = denoiser_config
if isinstance(denoiser_config, str):
with open(denoiser_config, "r") as f:
self.denoiser_config = json.load(f)
else:
self.denoiser_config = denoiser_config
|