Update app.py
Browse files
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 = "
|
| 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,
|