Spaces:

helamouri
/

medichat_assignment

Sleeping

App Files Files Community

medichat_assignment / src /__init__.py

helamouri

update model

eca6215 almost 2 years ago

Raw History Blame Contribute Delete

320 Bytes

from .config import *
from .model import load_model, configure_peft_model
from .dataset import load_and_prepare_dataset, formatting_prompts_func
from .training import train_model
from .inference import prepare_inference_inputs, generate_responses, stream_responses
from .save_model import save_model_and_tokenizer

	from .config import *
	from .model import load_model, configure_peft_model
	from .dataset import load_and_prepare_dataset, formatting_prompts_func
	from .training import train_model
	from .inference import prepare_inference_inputs, generate_responses, stream_responses
	from .save_model import save_model_and_tokenizer