sheldonl commited on
Commit
6d7cc64
·
1 Parent(s): eccdd96

Try fix ldd config

Browse files
Files changed (1) hide show
  1. world_generation_pipeline.py +1 -1
world_generation_pipeline.py CHANGED
@@ -266,7 +266,7 @@ class ARBaseGenerationPipeline(BaseWorldGenerationPipeline):
266
  """Load and initialize the diffusion decoder model."""
267
  self.diffusion_decoder_model = load_model_by_config(
268
  config_job_name=self.diffusion_decoder_config,
269
- config_file="cosmos1/models/autoregressive/diffusion_decoder/config/config_latent_diffusion_decoder.py",
270
  model_class=LatentDiffusionDecoderModel,
271
  )
272
  load_network_model(self.diffusion_decoder_model, self.diffusion_decoder_ckpt_path)
 
266
  """Load and initialize the diffusion decoder model."""
267
  self.diffusion_decoder_model = load_model_by_config(
268
  config_job_name=self.diffusion_decoder_config,
269
+ config_file="ldd_config.py",
270
  model_class=LatentDiffusionDecoderModel,
271
  )
272
  load_network_model(self.diffusion_decoder_model, self.diffusion_decoder_ckpt_path)