diff --git a/nerfstudio/models/dreamfusion.py b/nerfstudio/models/dreamfusion.py index b52c80ad4..136df8ffc 100644 --- a/nerfstudio/models/dreamfusion.py +++ b/nerfstudio/models/dreamfusion.py @@ -77,7 +77,7 @@ class DreamFusionModelConfig(ModelConfig): sphere_collider: bool = True """Use spherical collider instead of box""" target_transmittance_start: float = 0.8 - """target transmittance for opacity penalty. This is the percent of the scene that is + """target transmittance for opacity penalty. This is the percent of the scene that is background when rendered at the start of training""" target_transmittance_end: float = 0.4 """target transmittance for opacity penalty. This is the percent of the scene that is diff --git a/nerfstudio/pipelines/dreamfusion_pipeline.py b/nerfstudio/pipelines/dreamfusion_pipeline.py index 1bc12bef1..efbe14b65 100644 --- a/nerfstudio/pipelines/dreamfusion_pipeline.py +++ b/nerfstudio/pipelines/dreamfusion_pipeline.py @@ -137,7 +137,7 @@ class DreamfusionPipeline(VanillaPipeline): else: text_embedding = self.base_text_embedding - with torch.autocast(device_type='cuda', dtype=torch.float16): + with torch.autocast(device_type="cuda", dtype=torch.float16): sds_loss, latents, grad = self.sd.sds_loss( text_embedding.to(self.sd_device), train_output.to(self.sd_device),