INNER CODE UNIT · Python

mem_usage

kiri-art/docker-diffusers-api · api/app.py:711

        mem_usage = torch.cuda.memory_allocated() / torch.cuda.max_memory_allocated()

    result = result | {"$timings": getTimings(), "$mem_usage": mem_usage}

    return result

View source record →