app.py

@@ -58,7 +58,7 @@ model_cache_path = [elm for elm in Path(cache_loc).rglob("*") if Path(model_name
 logger.debug(f"{model_cache_path=}")
 
 if model_cache_path:
-    model_size_gb = Path(model_cache_path).stat().st_size / 2**30
+    model_size_gb = Path(model_cache_path[0]).stat().st_size / 2**30
     logger.info(f"{model_name=} {model_size_gb=:.2f} GB")
 
 # with gr.Blocks() as demo:

requirements.txt

@@ -2,5 +2,6 @@ torch
 transformers
 sentencepiece
 cpm_kernels
+accelerate
 
 loguru