automatic/modules/dml/optimizer/nvidia/__init__.py

from modules.dml.optimizer.optimizer import Optimizer

class nVidiaOptimizer(Optimizer):
    def memory_stats(index: int):
        # DML TODO: Implement or find a general (and also lightweight) way.
        return (1073741824, 0)