ffreemt commited on
Commit
200c2eb
1 Parent(s): 51add31
Files changed (1) hide show
  1. app.py +4 -3
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