Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -120,12 +120,12 @@ if use_colorfix:
|
|
| 120 |
from projects.video_diffusion_sr.color_fix import wavelet_reconstruction
|
| 121 |
|
| 122 |
def configure_runner():
|
| 123 |
-
config = load_config('
|
| 124 |
runner = VideoDiffusionInfer(config)
|
| 125 |
OmegaConf.set_readonly(runner.config, False)
|
| 126 |
# A chamada de inicialização crítica é feita aqui
|
| 127 |
init_torch(cudnn_benchmark=False, timeout=datetime.timedelta(seconds=3600))
|
| 128 |
-
runner.configure_dit_model(device="cuda", checkpoint='ckpts/
|
| 129 |
runner.configure_vae_model()
|
| 130 |
if hasattr(runner.vae, "set_memory_limit"):
|
| 131 |
runner.vae.set_memory_limit(**runner.config.vae.memory_limit)
|
|
|
|
| 120 |
from projects.video_diffusion_sr.color_fix import wavelet_reconstruction
|
| 121 |
|
| 122 |
def configure_runner():
|
| 123 |
+
config = load_config('configs_7b/main.yaml')
|
| 124 |
runner = VideoDiffusionInfer(config)
|
| 125 |
OmegaConf.set_readonly(runner.config, False)
|
| 126 |
# A chamada de inicialização crítica é feita aqui
|
| 127 |
init_torch(cudnn_benchmark=False, timeout=datetime.timedelta(seconds=3600))
|
| 128 |
+
runner.configure_dit_model(device="cuda", checkpoint='ckpts/seedvr2_ema_7b.pth')
|
| 129 |
runner.configure_vae_model()
|
| 130 |
if hasattr(runner.vae, "set_memory_limit"):
|
| 131 |
runner.vae.set_memory_limit(**runner.config.vae.memory_limit)
|