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

Upload 4 files

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -25,9 +25,11 @@ async def load_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...")
32
  base_model = AutoModelForCausalLM.from_pretrained(
33
  "mistralai/Mistral-7B-Instruct-v0.3",
 
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
+ # Debugging: Print the configuration
31
+ logger.info(f"Configuration: {config}")
32
+
33
  logger.info("Loading base model...")
34
  base_model = AutoModelForCausalLM.from_pretrained(
35
  "mistralai/Mistral-7B-Instruct-v0.3",