Spaces:
Runtime error
Runtime error
주영 commited on
Commit ·
4b450fe
1
Parent(s): 0ef76a2
modify kobart model path
Browse files- 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="
|
| 19 |
-
first_model = BartForConditionalGeneration.from_pretrained("jeju-potato/jeju_potato_models", subfolder="
|
| 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="
|
| 25 |
-
second_model = BartForConditionalGeneration.from_pretrained("jeju-potato/jeju_potato_models", subfolder="
|
| 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 |
|