ABIcode23 commited on
Commit
13002e8
1 Parent(s): 7353fdb

Upload 4 files

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -25,6 +25,7 @@ async def load_model():
25
  hf_token = os.environ.get("HUGGINGFACE_TOKEN")
26
 
27
  logger.info("Loading PEFT configuration...")
 
28
  config = PeftConfig.from_pretrained("frankmorales2020/Mistral-7B-text-to-sql-flash-attention-2-dataeval")
29
 
30
  logger.info("Loading base model...")
 
25
  hf_token = os.environ.get("HUGGINGFACE_TOKEN")
26
 
27
  logger.info("Loading PEFT configuration...")
28
+ # Ensure the config does not include unsupported arguments
29
  config = PeftConfig.from_pretrained("frankmorales2020/Mistral-7B-text-to-sql-flash-attention-2-dataeval")
30
 
31
  logger.info("Loading base model...")