remove total gpu memory

Author: sayakpaul

.github/workflows/benchmark.yml

@@ -46,7 +46,6 @@ jobs:
             HF_TOKEN: ${{ secrets.DIFFUSERS_BOT_TOKEN }}
             BASE_PATH: benchmark_outputs
         run: |
-          export TOTAL_GPU_MEMORY=$(python -c "import torch; print(torch.cuda.get_device_properties(0).total_memory / (1024**3))")
           cd benchmarks && python run_all.py && python push_results.py
           mkdir ${BASE_PATH} && mv *.csv ${BASE_PATH}
 

benchmarks/push_results.py

@@ -1,18 +1,8 @@
-import os
-
 import pandas as pd
-import torch
 from huggingface_hub import hf_hub_download, upload_file
 from huggingface_hub.utils import EntryNotFoundError
 
 
-if torch.cuda.is_available():
-    TOTAL_GPU_MEMORY = float(
-        os.getenv("TOTAL_GPU_MEMORY", torch.cuda.get_device_properties(0).total_memory / (1024**3))
-    )
-else:
-    raise
-
 REPO_ID = "diffusers/benchmarks"