Merge branch 'main' of github.com:timvieira/genparse

ChiSym · Jun 18, 2024 · a2fa116 · a2fa116
2 parents cc0a898 + bac6aa1
commit a2fa116
Show file tree

Hide file tree

Showing 29 changed files with 568 additions and 682 deletions.
diff --git a/bench/run_spider_llama2_chat.py b/bench/run_spider_llama2_chat.py
@@ -128,9 +128,9 @@ def main():
     model = 'meta-llama/Llama-2-7b-chat-hf'
     model = model.replace('7b', args.model_size)
     access_token = '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'
-    logger.info(f"using model {model}")
+    logger.info(f'using model {model}')
 
-    tokenizer = AutoTokenizer.from_pretrained(model, token=access_token)
+    # tokenizer = AutoTokenizer.from_pretrained(model, token=access_token)
     pipe = pipeline(
         'text-generation',
         model=model,

diff --git a/bench/spider/content_encoder.py b/bench/spider/content_encoder.py
@@ -46,7 +46,7 @@ def is_number(s: str) -> bool:
     try:
         float(s.replace(',', ''))
         return True
-    except:
+    except ValueError:
         return False