diff --git a/app.py b/app.py index 1e791bdd9c053eb9da4ffa6609651fd2662bc2ca..ca7470d1e621d66ca55cf63e5bcf0d0dd1bed7a4 100644 --- a/app.py +++ b/app.py @@ -1,10 +1,11 @@ import os import torch import random -import spaces +# import spaces import numpy as np import gradio as gr import soundfile as sf +from accelerate import Accelerator from transformers import T5Tokenizer, T5EncoderModel from diffusers import DDIMScheduler from src.models.conditioners import MaskDiT @@ -33,9 +34,12 @@ def load_models(config_name, ckpt_path, vae_path, device): unet.load_state_dict(torch.load(ckpt_path)['model']) unet.eval() + accelerator = Accelerator(mixed_precision="fp16") + unet = accelerator.prepare(unet) + # Load noise scheduler noise_scheduler = DDIMScheduler(**params['diff']) - + latents = torch.randn((1, 128, 128), device=device) noise = torch.randn_like(latents) timesteps = torch.randint(0, noise_scheduler.config.num_train_timesteps, (1,), device=device) @@ -43,6 +47,7 @@ def load_models(config_name, ckpt_path, vae_path, device): return autoencoder, unet, tokenizer, text_encoder, noise_scheduler, params + MAX_SEED = np.iinfo(np.int32).max # Model and config paths @@ -57,6 +62,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu' autoencoder, unet, tokenizer, text_encoder, noise_scheduler, params = load_models(config_name, ckpt_path, vae_path, device) + @spaces.GPU def generate_audio(text, length, guidance_scale, guidance_rescale, ddim_steps, eta, @@ -102,7 +108,7 @@ css = """ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo: with gr.Column(elem_id="col-container"): gr.Markdown(""" - # EzAudio Text-to-Audio Generator + # EzAudio: High-quality Text-to-Audio Generator Generate audio from text using a diffusion transformer. Adjust advanced settings for more control. """) @@ -125,10 +131,10 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo: with gr.Accordion("Advanced Settings", open=False): guidance_scale = gr.Slider(minimum=1.0, maximum=10, step=0.1, value=5.0, label="Guidance Scale") guidance_rescale = gr.Slider(minimum=0.0, maximum=1, step=0.05, value=0.75, label="Guidance Rescale") - ddim_steps = gr.Slider(minimum=25, maximum=200, step=5, value=100, label="DDIM Steps") + ddim_steps = gr.Slider(minimum=25, maximum=200, step=5, value=50, label="DDIM Steps") eta = gr.Slider(minimum=0.0, maximum=1.0, step=0.1, value=1.0, label="Eta") seed = gr.Slider(minimum=0, maximum=MAX_SEED, step=1, value=0, label="Seed") - randomize_seed = gr.Checkbox(label="Randomize Seed", value=False) + randomize_seed = gr.Checkbox(label="Randomize Seed (Disable Seed)", value=True) # Examples block gr.Examples( @@ -147,4 +153,4 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo: ) # Launch the Gradio demo -demo.launch() \ No newline at end of file +demo.launch() diff --git a/audiotools/__pycache__/__init__.cpython-310.pyc b/audiotools/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..77d9020459863ae8fd3c1fe56425ee12abdcba8d Binary files /dev/null and b/audiotools/__pycache__/__init__.cpython-310.pyc differ diff --git a/audiotools/core/__pycache__/__init__.cpython-310.pyc b/audiotools/core/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..f4da89d608dda2247a83a4d0028f08c1e05c7365 Binary files /dev/null and b/audiotools/core/__pycache__/__init__.cpython-310.pyc differ diff --git a/audiotools/core/__pycache__/audio_signal.cpython-310.pyc b/audiotools/core/__pycache__/audio_signal.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..174e92590630505ebd527f38fa73f52c810140a0 Binary files /dev/null and b/audiotools/core/__pycache__/audio_signal.cpython-310.pyc differ diff --git a/audiotools/core/__pycache__/display.cpython-310.pyc b/audiotools/core/__pycache__/display.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..13df7622e7084df474fb7999298c4734c401185d Binary files /dev/null and b/audiotools/core/__pycache__/display.cpython-310.pyc differ diff --git a/audiotools/core/__pycache__/dsp.cpython-310.pyc b/audiotools/core/__pycache__/dsp.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..41001a0bf20bd3470e53d7e20598d76ef0e04fa6 Binary files /dev/null and b/audiotools/core/__pycache__/dsp.cpython-310.pyc differ diff --git a/audiotools/core/__pycache__/effects.cpython-310.pyc b/audiotools/core/__pycache__/effects.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..b8571ce7d6e816a63e324ad7f09116fb1a63372c Binary files /dev/null and b/audiotools/core/__pycache__/effects.cpython-310.pyc differ diff --git a/audiotools/core/__pycache__/ffmpeg.cpython-310.pyc b/audiotools/core/__pycache__/ffmpeg.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..be5fa8d9f2bb4b78b7d53d4c55bc8e4b71063b4f Binary files /dev/null and b/audiotools/core/__pycache__/ffmpeg.cpython-310.pyc differ diff --git a/audiotools/core/__pycache__/loudness.cpython-310.pyc b/audiotools/core/__pycache__/loudness.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..9e9e9283e4881f660a21a1d991db7cce35fb6cdd Binary files /dev/null and b/audiotools/core/__pycache__/loudness.cpython-310.pyc differ diff --git a/audiotools/core/__pycache__/playback.cpython-310.pyc b/audiotools/core/__pycache__/playback.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..6455255d445232872592e92caf4f98e8ce0dc3b0 Binary files /dev/null and b/audiotools/core/__pycache__/playback.cpython-310.pyc differ diff --git a/audiotools/core/__pycache__/util.cpython-310.pyc b/audiotools/core/__pycache__/util.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..28edf75e56966ad4b3d8caa42889465e05b40370 Binary files /dev/null and b/audiotools/core/__pycache__/util.cpython-310.pyc differ diff --git a/audiotools/core/__pycache__/whisper.cpython-310.pyc b/audiotools/core/__pycache__/whisper.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..fb2cd1dd7ad8e3b827682392ce1e2c11d6073e00 Binary files /dev/null and b/audiotools/core/__pycache__/whisper.cpython-310.pyc differ diff --git a/audiotools/core/templates/__pycache__/__init__.cpython-310.pyc b/audiotools/core/templates/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..5265a456b28105747916c8a1dad89daa2632be04 Binary files /dev/null and b/audiotools/core/templates/__pycache__/__init__.cpython-310.pyc differ diff --git a/audiotools/data/__pycache__/__init__.cpython-310.pyc b/audiotools/data/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..ca11324f2bd8725389629adc708a992026b57667 Binary files /dev/null and b/audiotools/data/__pycache__/__init__.cpython-310.pyc differ diff --git a/audiotools/data/__pycache__/datasets.cpython-310.pyc b/audiotools/data/__pycache__/datasets.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..2a8488c6f76f15435c25078550ee9bfe1d3518fb Binary files /dev/null and b/audiotools/data/__pycache__/datasets.cpython-310.pyc differ diff --git a/audiotools/data/__pycache__/preprocess.cpython-310.pyc b/audiotools/data/__pycache__/preprocess.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..5b2ecc2f867c27415dd6f0732127925dd47f6777 Binary files /dev/null and b/audiotools/data/__pycache__/preprocess.cpython-310.pyc differ diff --git a/audiotools/data/__pycache__/transforms.cpython-310.pyc b/audiotools/data/__pycache__/transforms.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..d0673924c4a794331c7913b39f42a314c0c5dc73 Binary files /dev/null and b/audiotools/data/__pycache__/transforms.cpython-310.pyc differ diff --git a/audiotools/metrics/__pycache__/__init__.cpython-310.pyc b/audiotools/metrics/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..0c0373b3c569ac6309e1532251e2770b4feb126c Binary files /dev/null and b/audiotools/metrics/__pycache__/__init__.cpython-310.pyc differ diff --git a/audiotools/metrics/__pycache__/distance.cpython-310.pyc b/audiotools/metrics/__pycache__/distance.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..7106850dcd6fcd6456cfc686e08bff1d6df28576 Binary files /dev/null and b/audiotools/metrics/__pycache__/distance.cpython-310.pyc differ diff --git a/audiotools/metrics/__pycache__/quality.cpython-310.pyc b/audiotools/metrics/__pycache__/quality.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..8c8bd6d8268ab191ae360688002bb9cf5ab535b8 Binary files /dev/null and b/audiotools/metrics/__pycache__/quality.cpython-310.pyc differ diff --git a/audiotools/metrics/__pycache__/spectral.cpython-310.pyc b/audiotools/metrics/__pycache__/spectral.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..ee3b7960c30d963c884d182f1e0bdac8980db5e8 Binary files /dev/null and b/audiotools/metrics/__pycache__/spectral.cpython-310.pyc differ diff --git a/audiotools/ml/__pycache__/__init__.cpython-310.pyc b/audiotools/ml/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..58cea77971d01badeaacb9055e28b7eda929d588 Binary files /dev/null and b/audiotools/ml/__pycache__/__init__.cpython-310.pyc differ diff --git a/audiotools/ml/__pycache__/accelerator.cpython-310.pyc b/audiotools/ml/__pycache__/accelerator.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..0bb83d2cdd9c6c38084cc8b0287ffa0be3f322cc Binary files /dev/null and b/audiotools/ml/__pycache__/accelerator.cpython-310.pyc differ diff --git a/audiotools/ml/__pycache__/decorators.cpython-310.pyc b/audiotools/ml/__pycache__/decorators.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..8ffd4df8ab76bc25d5ccefaae8f2785d2e8748b9 Binary files /dev/null and b/audiotools/ml/__pycache__/decorators.cpython-310.pyc differ diff --git a/audiotools/ml/__pycache__/experiment.cpython-310.pyc b/audiotools/ml/__pycache__/experiment.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..81f4dfca76b18694ef1bd18dfbcb06c218c50a18 Binary files /dev/null and b/audiotools/ml/__pycache__/experiment.cpython-310.pyc differ diff --git a/audiotools/ml/layers/__pycache__/__init__.cpython-310.pyc b/audiotools/ml/layers/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..5b241bd603b69b876e3a971f691b586c2f907751 Binary files /dev/null and b/audiotools/ml/layers/__pycache__/__init__.cpython-310.pyc differ diff --git a/audiotools/ml/layers/__pycache__/base.cpython-310.pyc b/audiotools/ml/layers/__pycache__/base.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..fc0b4972f929abcaea92d1b575a17d93a3478d7e Binary files /dev/null and b/audiotools/ml/layers/__pycache__/base.cpython-310.pyc differ diff --git a/audiotools/ml/layers/__pycache__/spectral_gate.cpython-310.pyc b/audiotools/ml/layers/__pycache__/spectral_gate.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..63c90a2d10d30be152e629d6ec379be1ae7d6667 Binary files /dev/null and b/audiotools/ml/layers/__pycache__/spectral_gate.cpython-310.pyc differ diff --git a/src/__pycache__/inference.cpython-310.pyc b/src/__pycache__/inference.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..50cea45281578d634bef0e6894f461290bc6d0b4 Binary files /dev/null and b/src/__pycache__/inference.cpython-310.pyc differ diff --git a/src/models/__pycache__/blocks.cpython-310.pyc b/src/models/__pycache__/blocks.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..17e7048c5c700487de51acaa6b5d546002bf8c5a Binary files /dev/null and b/src/models/__pycache__/blocks.cpython-310.pyc differ diff --git a/src/models/__pycache__/conditioners.cpython-310.pyc b/src/models/__pycache__/conditioners.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..731fecbb309f93a5a821b6b34a20fdf74fa3b5a0 Binary files /dev/null and b/src/models/__pycache__/conditioners.cpython-310.pyc differ diff --git a/src/models/__pycache__/udit.cpython-310.pyc b/src/models/__pycache__/udit.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..c2d88049e1e885f4aaf2b2534ad931f8bfbf618d Binary files /dev/null and b/src/models/__pycache__/udit.cpython-310.pyc differ diff --git a/src/models/utils/__pycache__/__init__.cpython-310.pyc b/src/models/utils/__pycache__/__init__.cpython-310.pyc index fc31b0a19bf83eef2a703df69b4272a12bfbe577..e75e47b1d25d7dec7dcf7ad8ed1aeca588f02054 100644 Binary files a/src/models/utils/__pycache__/__init__.cpython-310.pyc and b/src/models/utils/__pycache__/__init__.cpython-310.pyc differ diff --git a/src/models/utils/__pycache__/attention.cpython-310.pyc b/src/models/utils/__pycache__/attention.cpython-310.pyc index 9ebbb2fc665b2c2ff2115bc98ec054430333e6ee..dc880d8a7367ff0d637ccfb1f797eef233574081 100644 Binary files a/src/models/utils/__pycache__/attention.cpython-310.pyc and b/src/models/utils/__pycache__/attention.cpython-310.pyc differ diff --git a/src/models/utils/__pycache__/modules.cpython-310.pyc b/src/models/utils/__pycache__/modules.cpython-310.pyc index d507df3df9cbf9fa29fbabb4591df36aedf6bdd4..66b0e11ece38a64070f4ec7669a4e446f7b16a1f 100644 Binary files a/src/models/utils/__pycache__/modules.cpython-310.pyc and b/src/models/utils/__pycache__/modules.cpython-310.pyc differ diff --git a/src/models/utils/__pycache__/rotary.cpython-310.pyc b/src/models/utils/__pycache__/rotary.cpython-310.pyc index 716bfd4abc834b3e912c4f4574ddc3d3597183a5..f4ee88d28fa98082fd51535d702dd9f4c6e69d46 100644 Binary files a/src/models/utils/__pycache__/rotary.cpython-310.pyc and b/src/models/utils/__pycache__/rotary.cpython-310.pyc differ diff --git a/src/models/utils/__pycache__/span_mask.cpython-310.pyc b/src/models/utils/__pycache__/span_mask.cpython-310.pyc index 8dc66ee9fc445b41e939f29ba32cfdeb6169bcc2..6db5df6b6f7e97ced70c65caeb4daa3826203a1b 100644 Binary files a/src/models/utils/__pycache__/span_mask.cpython-310.pyc and b/src/models/utils/__pycache__/span_mask.cpython-310.pyc differ diff --git a/src/models/utils/__pycache__/timm.cpython-310.pyc b/src/models/utils/__pycache__/timm.cpython-310.pyc index 0fa38c2a3330015ecde142ac1e187afd6afd3aa5..5eed0419dbf42d1d97e6b40a71be0e815ceeecab 100644 Binary files a/src/models/utils/__pycache__/timm.cpython-310.pyc and b/src/models/utils/__pycache__/timm.cpython-310.pyc differ diff --git a/src/modules/__pycache__/autoencoder_wrapper.cpython-310.pyc b/src/modules/__pycache__/autoencoder_wrapper.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..88662a062b0d61b39d8d722dc27fd91194d91a75 Binary files /dev/null and b/src/modules/__pycache__/autoencoder_wrapper.cpython-310.pyc differ diff --git a/src/modules/dac/__pycache__/__init__.cpython-310.pyc b/src/modules/dac/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..094b9a491ba57d6b27cc5848cf58e47b9f9d8a46 Binary files /dev/null and b/src/modules/dac/__pycache__/__init__.cpython-310.pyc differ diff --git a/src/modules/dac/model/__pycache__/__init__.cpython-310.pyc b/src/modules/dac/model/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..edd0356d0ec9a8ad5e243b38859d7768fc36ccc3 Binary files /dev/null and b/src/modules/dac/model/__pycache__/__init__.cpython-310.pyc differ diff --git a/src/modules/dac/model/__pycache__/base.cpython-310.pyc b/src/modules/dac/model/__pycache__/base.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..f9add97b61d826dd1976a5b39f98dbad921f30c9 Binary files /dev/null and b/src/modules/dac/model/__pycache__/base.cpython-310.pyc differ diff --git a/src/modules/dac/model/__pycache__/dac.cpython-310.pyc b/src/modules/dac/model/__pycache__/dac.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..f0f3b022b32a3050dd63db52e52cf48762b2ce54 Binary files /dev/null and b/src/modules/dac/model/__pycache__/dac.cpython-310.pyc differ diff --git a/src/modules/dac/model/__pycache__/discriminator.cpython-310.pyc b/src/modules/dac/model/__pycache__/discriminator.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..c8030b5b968c0c8dbd34b07e2565ebca8ea53eeb Binary files /dev/null and b/src/modules/dac/model/__pycache__/discriminator.cpython-310.pyc differ diff --git a/src/modules/dac/nn/__pycache__/__init__.cpython-310.pyc b/src/modules/dac/nn/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..cb7379d45597bfa7f3849a82e582a138c640eb79 Binary files /dev/null and b/src/modules/dac/nn/__pycache__/__init__.cpython-310.pyc differ diff --git a/src/modules/dac/nn/__pycache__/layers.cpython-310.pyc b/src/modules/dac/nn/__pycache__/layers.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..333fe1704e49396fb22064d0e8fe21866cc7c3ea Binary files /dev/null and b/src/modules/dac/nn/__pycache__/layers.cpython-310.pyc differ diff --git a/src/modules/dac/nn/__pycache__/loss.cpython-310.pyc b/src/modules/dac/nn/__pycache__/loss.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..14191fd8de65948f42f5c1baf229bed6a284cb24 Binary files /dev/null and b/src/modules/dac/nn/__pycache__/loss.cpython-310.pyc differ diff --git a/src/modules/dac/nn/__pycache__/quantize.cpython-310.pyc b/src/modules/dac/nn/__pycache__/quantize.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..7e1a654f113c4df49e4bc9073204b2c4edcd883e Binary files /dev/null and b/src/modules/dac/nn/__pycache__/quantize.cpython-310.pyc differ diff --git a/src/modules/dac/utils/__pycache__/__init__.cpython-310.pyc b/src/modules/dac/utils/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..a8924cb8604a54811ffbe7d28adc347f6f26f2fb Binary files /dev/null and b/src/modules/dac/utils/__pycache__/__init__.cpython-310.pyc differ diff --git a/src/modules/stable_vae/__pycache__/__init__.cpython-310.pyc b/src/modules/stable_vae/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..2e52803a886cb151d21f2c9ca72a7022033e4f34 Binary files /dev/null and b/src/modules/stable_vae/__pycache__/__init__.cpython-310.pyc differ diff --git a/src/modules/stable_vae/models/__pycache__/autoencoders.cpython-310.pyc b/src/modules/stable_vae/models/__pycache__/autoencoders.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..8056a78fef615545b643282845811977981e0a14 Binary files /dev/null and b/src/modules/stable_vae/models/__pycache__/autoencoders.cpython-310.pyc differ diff --git a/src/modules/stable_vae/models/__pycache__/blocks.cpython-310.pyc b/src/modules/stable_vae/models/__pycache__/blocks.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..608be76a3f4713e4c98d81db66ccb97ba0658f30 Binary files /dev/null and b/src/modules/stable_vae/models/__pycache__/blocks.cpython-310.pyc differ diff --git a/src/modules/stable_vae/models/__pycache__/bottleneck.cpython-310.pyc b/src/modules/stable_vae/models/__pycache__/bottleneck.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..939b2322d642d0dfd2f7ae1487b2e9acdd2bcee8 Binary files /dev/null and b/src/modules/stable_vae/models/__pycache__/bottleneck.cpython-310.pyc differ diff --git a/src/modules/stable_vae/models/__pycache__/factory.cpython-310.pyc b/src/modules/stable_vae/models/__pycache__/factory.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..b2c9966d9f263be2f684845cf7a9dc74336cdaf1 Binary files /dev/null and b/src/modules/stable_vae/models/__pycache__/factory.cpython-310.pyc differ diff --git a/src/modules/stable_vae/models/__pycache__/pretransforms.cpython-310.pyc b/src/modules/stable_vae/models/__pycache__/pretransforms.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..8cb2d733410527b0df9ad3ed8dccd42eeb30ac90 Binary files /dev/null and b/src/modules/stable_vae/models/__pycache__/pretransforms.cpython-310.pyc differ diff --git a/src/modules/stable_vae/models/__pycache__/utils.cpython-310.pyc b/src/modules/stable_vae/models/__pycache__/utils.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..768c594244e513599068cc966b5c493884842c6d Binary files /dev/null and b/src/modules/stable_vae/models/__pycache__/utils.cpython-310.pyc differ diff --git a/src/modules/stable_vae/models/nn/__pycache__/__init__.cpython-310.pyc b/src/modules/stable_vae/models/nn/__pycache__/__init__.cpython-310.pyc index fd83e18bd22222ca6b9ce0f0ab056cc026747bb3..efe8b7df217dd8f2d17ae8cd789528c1583db399 100644 Binary files a/src/modules/stable_vae/models/nn/__pycache__/__init__.cpython-310.pyc and b/src/modules/stable_vae/models/nn/__pycache__/__init__.cpython-310.pyc differ diff --git a/src/modules/stable_vae/models/nn/__pycache__/layers.cpython-310.pyc b/src/modules/stable_vae/models/nn/__pycache__/layers.cpython-310.pyc index 57a85d0e28029662f1ecc2790e44f71caa09cd0c..50d77963c7bf791ce636e8b9f3bc1111f3f7db85 100644 Binary files a/src/modules/stable_vae/models/nn/__pycache__/layers.cpython-310.pyc and b/src/modules/stable_vae/models/nn/__pycache__/layers.cpython-310.pyc differ diff --git a/src/modules/stable_vae/models/nn/__pycache__/loss.cpython-310.pyc b/src/modules/stable_vae/models/nn/__pycache__/loss.cpython-310.pyc index 47d3c0daa35a755146c3ebf4f49d430469fe0c6a..676f0af69b1ac293343d66dddf026dc6313e4f09 100644 Binary files a/src/modules/stable_vae/models/nn/__pycache__/loss.cpython-310.pyc and b/src/modules/stable_vae/models/nn/__pycache__/loss.cpython-310.pyc differ diff --git a/src/modules/stable_vae/models/nn/__pycache__/quantize.cpython-310.pyc b/src/modules/stable_vae/models/nn/__pycache__/quantize.cpython-310.pyc index d960f1c6e8f02f6b1c3b72b9d60543ceadb619cf..b9f4162124bb1075d89de96c68397a3508bab20e 100644 Binary files a/src/modules/stable_vae/models/nn/__pycache__/quantize.cpython-310.pyc and b/src/modules/stable_vae/models/nn/__pycache__/quantize.cpython-310.pyc differ diff --git a/src/utils/__pycache__/__init__.cpython-310.pyc b/src/utils/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..49974681b9d27ca5a286a8b9c612cc886b843219 Binary files /dev/null and b/src/utils/__pycache__/__init__.cpython-310.pyc differ diff --git a/src/utils/__pycache__/utils.cpython-310.pyc b/src/utils/__pycache__/utils.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..f2af21e024d10a0b596e9927979e25d88ee6b1a9 Binary files /dev/null and b/src/utils/__pycache__/utils.cpython-310.pyc differ