Aduc-sdr commited on
Commit
a116766
·
verified ·
1 Parent(s): 6c1128d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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('configs_3b/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_3b.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)
 
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)