Update inference_test_turkcell_with_intents.py
Browse files
inference_test_turkcell_with_intents.py
CHANGED
@@ -3,7 +3,7 @@ import os, torch, threading, uvicorn, time, traceback, zipfile, random, json, sh
|
|
3 |
from fastapi import FastAPI
|
4 |
from fastapi.responses import HTMLResponse, JSONResponse
|
5 |
from pydantic import BaseModel
|
6 |
-
from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequenceClassification, Trainer, TrainingArguments, default_data_collator
|
7 |
from peft import PeftModel
|
8 |
from datasets import Dataset
|
9 |
from datetime import datetime
|
|
|
3 |
from fastapi import FastAPI
|
4 |
from fastapi.responses import HTMLResponse, JSONResponse
|
5 |
from pydantic import BaseModel
|
6 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequenceClassification, Trainer, TrainingArguments, default_data_collator, AutoConfig
|
7 |
from peft import PeftModel
|
8 |
from datasets import Dataset
|
9 |
from datetime import datetime
|