Files
automatic/modules/dml/optimizer/amd/__init__.py
T
2023-04-26 17:44:32 +09:00

8 lines
294 B
Python

from modules.dml.optimizer.optimizer import Optimizer
from .driver.atiadlxx import ATIADLxx
class AMDOptimizer(Optimizer):
driver: ATIADLxx = ATIADLxx()
def memory_stats(index):
return (AMDOptimizer.driver.iHyperMemorySize, AMDOptimizer.driver.get_dedicated_vram_usage(index))