Spaces:
Runtime error
Runtime error
ffreemt
commited on
Commit
•
200c2eb
1
Parent(s):
51add31
app.py
CHANGED
@@ -50,10 +50,11 @@ logger.debug("done load")
|
|
50 |
# tokenizer = AutoTokenizer.from_pretrained("openchat/openchat_v2_w")
|
51 |
# model = AutoModelForCausalLM.from_pretrained("openchat/openchat_v2_w", load_in_8bit_fp32_cpu_offload=True, load_in_8bit=True)
|
52 |
|
53 |
-
model_path = model.config._dict["model_name_or_path"]
|
54 |
-
logger.debug(f"{model.config=} {type(model.config)=} {model_path=}")
|
|
|
55 |
|
56 |
-
model_size_gb = Path(model_path).stat().st_size / 2**30
|
57 |
|
58 |
logger.info(f"{model_name=} {model_size_gb=:.2f} GB")
|
59 |
|
|
|
50 |
# tokenizer = AutoTokenizer.from_pretrained("openchat/openchat_v2_w")
|
51 |
# model = AutoModelForCausalLM.from_pretrained("openchat/openchat_v2_w", load_in_8bit_fp32_cpu_offload=True, load_in_8bit=True)
|
52 |
|
53 |
+
# model_path = model.config._dict["model_name_or_path"]
|
54 |
+
# logger.debug(f"{model.config=} {type(model.config)=} {model_path=}")
|
55 |
+
logger.debug(f"{model.config=}, {type(model.config)=} ")
|
56 |
|
57 |
+
# model_size_gb = Path(model_path).stat().st_size / 2**30
|
58 |
|
59 |
logger.info(f"{model_name=} {model_size_gb=:.2f} GB")
|
60 |
|