AliEssa555 commited on
Commit
a544069
·
verified ·
1 Parent(s): e1859d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/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/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)
 
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)