Lint

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

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),