choco9966 commited on
Commit
4f7633e
1 Parent(s): 8d119d2

[Add] load_from_hub

Browse files
Files changed (1) hide show
  1. src/load_from_hub.py +1 -0
src/load_from_hub.py CHANGED
@@ -136,6 +136,7 @@ def get_evaluation_queue_df(
136
 
137
 
138
  def is_model_on_hub(model_name: str, revision: str) -> bool:
 
139
  try:
140
  AutoConfig.from_pretrained(model_name, revision=revision, trust_remote_code=False)
141
  return True, None
 
136
 
137
 
138
  def is_model_on_hub(model_name: str, revision: str) -> bool:
139
+ print("is_model_on_hub", model_name, revision)
140
  try:
141
  AutoConfig.from_pretrained(model_name, revision=revision, trust_remote_code=False)
142
  return True, None