Commit 6ced7f8

committed

Add the trainer Class for train the Model.

1 parent f8756fe commit 6ced7f8Copy full SHA for 6ced7f8

File tree

-5

lines changed

-5

lines changed

Lines changed: 5 additions & 5 deletions

Original file line number	Diff line number	Diff line change
`@@ -5,11 +5,11 @@`
`5`	`5`	`from pathlib import Path`
`6`	`6`	`from tqdm import tqdm`
`7`	`7`
`8`		`-from model.transformers_model import DiffusionLLM`
`9`		`-from model.mask_token import mask_tokens_for_diffusion`
`10`		`-from evaluate import evaluate`
`11`		`-from model_save import save_model`
`12`		`-from utils.error_handler import handle_errors, DiffusionLMError`
	`8`	`+from diffusionLM.model.transformers_model import DiffusionLLM`
	`9`	`+from diffusionLM.model.mask_token import mask_tokens_for_diffusion`
	`10`	`+from .evaluate import evaluate`
	`11`	`+from diffusionLM.save_model.model_save import save_model`
	`12`	`+from diffusionLM.utils.error_handler import handle_errors, DiffusionLMError`
`13`	`13`
`14`	`14`	`logger = logging.getLogger(__name__)`
`15`	`15`

Comments

(0)