주영 commited on
Commit
4b450fe
·
1 Parent(s): 0ef76a2

modify kobart model path

Browse files
Files changed (1) hide show
  1. app/inference.py +4 -4
app/inference.py CHANGED
@@ -15,14 +15,14 @@ whisper_model.config.forced_decoder_ids = None
15
  whisper_model.generation_config.forced_decoder_ids = None
16
 
17
  # first_kobart 로딩
18
- first_tokenizer = PreTrainedTokenizerFast.from_pretrained("jeju-potato/jeju_potato_models", subfolder="first_kobart_v3")
19
- first_model = BartForConditionalGeneration.from_pretrained("jeju-potato/jeju_potato_models", subfolder="first_kobart_v3").to(device)
20
  first_model.config.forced_decoder_ids = None
21
  first_model.generation_config.forced_decoder_ids = None
22
 
23
  # second_kobart 로딩
24
- second_tokenizer = PreTrainedTokenizerFast.from_pretrained("jeju-potato/jeju_potato_models", subfolder="second_kobart_v2")
25
- second_model = BartForConditionalGeneration.from_pretrained("jeju-potato/jeju_potato_models", subfolder="second_kobart_v2").to(device)
26
  second_model.config.forced_decoder_ids = None
27
  second_model.generation_config.forced_decoder_ids = None
28
 
 
15
  whisper_model.generation_config.forced_decoder_ids = None
16
 
17
  # first_kobart 로딩
18
+ first_tokenizer = PreTrainedTokenizerFast.from_pretrained("jeju-potato/jeju_potato_models", subfolder="first_kobart_finetuned_v3")
19
+ first_model = BartForConditionalGeneration.from_pretrained("jeju-potato/jeju_potato_models", subfolder="first_kobart_finetuned_v3").to(device)
20
  first_model.config.forced_decoder_ids = None
21
  first_model.generation_config.forced_decoder_ids = None
22
 
23
  # second_kobart 로딩
24
+ second_tokenizer = PreTrainedTokenizerFast.from_pretrained("jeju-potato/jeju_potato_models", subfolder="second_kobart_finetuned_v2")
25
+ second_model = BartForConditionalGeneration.from_pretrained("jeju-potato/jeju_potato_models", subfolder="second_kobart_finetuned_v2").to(device)
26
  second_model.config.forced_decoder_ids = None
27
  second_model.generation_config.forced_decoder_ids = None
28