Julian Bilcke
commited on
Commit
·
e386f72
1
Parent(s):
dd52c86
lol, woops
Browse files- vms/training_service.py +2 -1
vms/training_service.py
CHANGED
|
@@ -18,7 +18,8 @@ import select
|
|
| 18 |
|
| 19 |
from typing import Any, Optional, Dict, List, Union, Tuple
|
| 20 |
|
| 21 |
-
from
|
|
|
|
| 22 |
from .config import TrainingConfig, TRAINING_PRESETS, LOG_FILE_PATH, TRAINING_VIDEOS_PATH, STORAGE_PATH, TRAINING_PATH, MODEL_PATH, OUTPUT_PATH, HF_API_TOKEN, MODEL_TYPES
|
| 23 |
from .utils import make_archive, parse_training_log, is_image_file, is_video_file
|
| 24 |
from .finetrainers_utils import prepare_finetrainers_dataset, copy_files_to_training_dir
|
|
|
|
| 18 |
|
| 19 |
from typing import Any, Optional, Dict, List, Union, Tuple
|
| 20 |
|
| 21 |
+
from huggingface_hub import upload_folder, create_repo
|
| 22 |
+
|
| 23 |
from .config import TrainingConfig, TRAINING_PRESETS, LOG_FILE_PATH, TRAINING_VIDEOS_PATH, STORAGE_PATH, TRAINING_PATH, MODEL_PATH, OUTPUT_PATH, HF_API_TOKEN, MODEL_TYPES
|
| 24 |
from .utils import make_archive, parse_training_log, is_image_file, is_video_file
|
| 25 |
from .finetrainers_utils import prepare_finetrainers_dataset, copy_files_to_training_dir
|