Try fix ldd config
Browse files
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="
|
| 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)
|