raygiles3 commited on
Commit
d68aedf
·
verified ·
1 Parent(s): 14ee4cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-base")
19
  whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-base")
20
 
21
  # Initialize the summarization model and tokenizer
22
- # Load LLAMA 7B model with accelerate
23
- model_name = "/Llama-2-7b-hf"
24
  with init_empty_weights():
25
  summarization_model = AutoModelForCausalLM.from_pretrained(model_name)
26
 
 
19
  whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-base")
20
 
21
  # Initialize the summarization model and tokenizer
22
+ # Load LLAMA 7B model with accelerate from local path
23
+ model_name = "./llama-2-7b-hf"
24
  with init_empty_weights():
25
  summarization_model = AutoModelForCausalLM.from_pretrained(model_name)
26