Upload 4 files
Browse files
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...")
|