Spaces:
Runtime error
Runtime error
ffreemt
commited on
Commit
•
7eec20e
1
Parent(s):
84807aa
- app.py +1 -1
- requirements.txt +1 -0
app.py
CHANGED
@@ -58,7 +58,7 @@ model_cache_path = [elm for elm in Path(cache_loc).rglob("*") if Path(model_name
|
|
58 |
logger.debug(f"{model_cache_path=}")
|
59 |
|
60 |
if model_cache_path:
|
61 |
-
model_size_gb = Path(model_cache_path).stat().st_size / 2**30
|
62 |
logger.info(f"{model_name=} {model_size_gb=:.2f} GB")
|
63 |
|
64 |
# with gr.Blocks() as demo:
|
|
|
58 |
logger.debug(f"{model_cache_path=}")
|
59 |
|
60 |
if model_cache_path:
|
61 |
+
model_size_gb = Path(model_cache_path[0]).stat().st_size / 2**30
|
62 |
logger.info(f"{model_name=} {model_size_gb=:.2f} GB")
|
63 |
|
64 |
# with gr.Blocks() as demo:
|
requirements.txt
CHANGED
@@ -2,5 +2,6 @@ torch
|
|
2 |
transformers
|
3 |
sentencepiece
|
4 |
cpm_kernels
|
|
|
5 |
|
6 |
loguru
|
|
|
2 |
transformers
|
3 |
sentencepiece
|
4 |
cpm_kernels
|
5 |
+
accelerate
|
6 |
|
7 |
loguru
|