zavavan commited on
Commit
c228041
·
verified ·
1 Parent(s): dc0ac33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -359,7 +359,7 @@ def load_model(model_key):
359
  custom_llm = CustomHuggingFaceGenerator(model=model_instance, tokenizer=tokenizer, default_generation_kwargs=generation_parameters)
360
 
361
  else:
362
- model_path = "/content/drive/MyDrive/AI-DORE_experiments/phi-2-personaLoraBuilder" # replace with actual path
363
  base_model_id = "microsoft/phi-2"
364
  bnb_config = BitsAndBytesConfig(
365
  load_in_4bit=True,
 
359
  custom_llm = CustomHuggingFaceGenerator(model=model_instance, tokenizer=tokenizer, default_generation_kwargs=generation_parameters)
360
 
361
  else:
362
+ model_path = "./phi-2-personaLoraBuilder" # replace with actual path
363
  base_model_id = "microsoft/phi-2"
364
  bnb_config = BitsAndBytesConfig(
365
  load_in_4bit=True,