Spaces:
Runtime error
Runtime error
AliEssa555
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -6,9 +6,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
6 |
from peft import PeftModel, PeftConfig
|
7 |
from transformers import AutoModelForCausalLM
|
8 |
|
9 |
-
config = PeftConfig.from_pretrained("AliEssa555/
|
10 |
base_model = AutoModelForCausalLM.from_pretrained("TheBloke/Mistral-7B-Instruct-v0.2-GPTQ")
|
11 |
-
model_name = PeftModel.from_pretrained(base_model, "AliEssa555/
|
12 |
|
13 |
#model_name = "path_to_your_fine_tuned_model" # Use the local path or the Hugging Face model hub ID if published
|
14 |
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16)
|
|
|
6 |
from peft import PeftModel, PeftConfig
|
7 |
from transformers import AutoModelForCausalLM
|
8 |
|
9 |
+
config = PeftConfig.from_pretrained("AliEssa555/latest-podcast-model-ft")
|
10 |
base_model = AutoModelForCausalLM.from_pretrained("TheBloke/Mistral-7B-Instruct-v0.2-GPTQ")
|
11 |
+
model_name = PeftModel.from_pretrained(base_model, "AliEssa555/latest-podcast-model-ft")
|
12 |
|
13 |
#model_name = "path_to_your_fine_tuned_model" # Use the local path or the Hugging Face model hub ID if published
|
14 |
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16)
|