From b79aa04b620c796c7eb2d2610dbe5dbdf448b3ac Mon Sep 17 00:00:00 2001 From: Matt Tancik Date: Tue, 14 Feb 2023 10:39:45 -0800 Subject: [PATCH] Cleanup --- nerfstudio/pipelines/dreamfusion_pipeline.py | 18 ------------------ 1 file changed, 18 deletions(-) diff --git a/nerfstudio/pipelines/dreamfusion_pipeline.py b/nerfstudio/pipelines/dreamfusion_pipeline.py index 6d29cb0f0..3123bfe70 100644 --- a/nerfstudio/pipelines/dreamfusion_pipeline.py +++ b/nerfstudio/pipelines/dreamfusion_pipeline.py @@ -17,10 +17,7 @@ from dataclasses import dataclass, field from typing import Optional, Type -import matplotlib.pyplot as plt -import numpy as np import torch -from torch.cuda.amp import custom_bwd, custom_fwd from torch.cuda.amp.grad_scaler import GradScaler from typing_extensions import Literal @@ -31,21 +28,6 @@ from nerfstudio.generative.stable_diffusion import StableDiffusion from nerfstudio.pipelines.base_pipeline import VanillaPipeline, VanillaPipelineConfig -class SpecifyGradient(torch.autograd.Function): - @staticmethod - @custom_fwd - def forward(ctx, input_tensor, gt_grad): - ctx.save_for_backward(gt_grad) - return torch.zeros([], device=input_tensor.device, dtype=input_tensor.dtype) - - @staticmethod - @custom_bwd - def backward(ctx, grad): - (gt_grad,) = ctx.saved_tensors - batch_size = len(gt_grad) - return gt_grad / batch_size, None - - @dataclass class DreamfusionPipelineConfig(VanillaPipelineConfig): """Configuration for pipeline instantiation"""