Skip to content

Commit 6ed8bbd

Browse files
committed
Formatting
1 parent 114ccf7 commit 6ed8bbd

File tree

1 file changed

+2
-19
lines changed

1 file changed

+2
-19
lines changed

tests/models.test.js

Lines changed: 2 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -2,24 +2,7 @@
22
* Test that models loaded outside of the `pipeline` function work correctly (e.g., `AutoModel.from_pretrained(...)`);
33
*/
44

5-
import {
6-
AutoTokenizer,
7-
AutoProcessor,
8-
BertForMaskedLM,
9-
GPT2LMHeadModel,
10-
T5ForConditionalGeneration,
11-
BertTokenizer,
12-
GPT2Tokenizer,
13-
T5Tokenizer,
14-
LlamaTokenizer,
15-
LlamaForCausalLM,
16-
WhisperForConditionalGeneration,
17-
WhisperProcessor,
18-
AutoModelForMaskedLM,
19-
AutoModelForCausalLM,
20-
AutoModelForSeq2SeqLM,
21-
AutoModelForSpeechSeq2Seq,
22-
} from "../src/transformers.js";
5+
import { AutoTokenizer, AutoProcessor, BertForMaskedLM, GPT2LMHeadModel, T5ForConditionalGeneration, BertTokenizer, GPT2Tokenizer, T5Tokenizer, LlamaTokenizer, LlamaForCausalLM, WhisperForConditionalGeneration, WhisperProcessor, AutoModelForMaskedLM, AutoModelForCausalLM, AutoModelForSeq2SeqLM, AutoModelForSpeechSeq2Seq } from "../src/transformers.js";
236
import { init, MAX_TEST_EXECUTION_TIME, DEFAULT_MODEL_OPTIONS } from "./init.js";
247
import { compare, collect_and_execute_tests } from "./test_utils.js";
258

@@ -55,7 +38,7 @@ describe("Loading different architecture types", () => {
5538
const tests = [
5639
texts[0], // single
5740
texts, // batched
58-
]
41+
];
5942

6043
const { model_type } = model.config;
6144
const tokenizer = model_type === "whisper" ? processor.tokenizer : processor;

0 commit comments

Comments
 (0)