changing source directory name

Author: benjaminye

src/__init__.py llmtune/__init__.pysrc/__init__.py renamed to llmtune/__init__.py

@@ -6,11 +6,8 @@
 
 import re
 from datasets import Dataset
-from rich.console import Console
-from rich.layout import Layout
-from rich.panel import Panel
 
-from src.data.ingestor import Ingestor, get_ingestor
+from llmtune.data.ingestor import Ingestor, get_ingestor
 
 
 class DatasetGenerator:

src/data/__init__.py llmtune/data/__init__.pysrc/data/__init__.py renamed to llmtune/data/__init__.py

@@ -1,5 +1,4 @@
 from abc import ABC, abstractmethod
-from typing import Union, List, Tuple, Dict
 
 
 class Finetune(ABC):

src/data/dataset_generator.py llmtune/data/dataset_generator.pysrc/data/dataset_generator.py renamed to llmtune/data/dataset_generator.py

@@ -5,7 +5,6 @@
 
 import bitsandbytes as bnb
 from datasets import Dataset
-from accelerate import Accelerator
 from transformers import (
     AutoTokenizer,
     AutoModelForCausalLM,
@@ -23,10 +22,10 @@
 from rich.console import Console
 
 
-from src.pydantic_models.config_model import Config
-from src.utils.save_utils import DirectoryHelper
-from src.finetune.finetune import Finetune
-from src.ui.rich_ui import RichUI
+from llmtune.pydantic_models.config_model import Config
+from llmtune.utils.save_utils import DirectoryHelper
+from llmtune.finetune.generics import Finetune
+from llmtune.ui.rich_ui import RichUI
 
 
 class LoRAFinetune(Finetune):

src/data/ingestor.py llmtune/data/ingestor.pysrc/data/ingestor.py renamed to llmtune/data/ingestor.py

@@ -1,5 +1,4 @@
 from abc import ABC, abstractmethod
-from typing import Union, List, Tuple, Dict
 
 
 class Inference(ABC):

src/finetune/__init__.py llmtune/finetune/__init__.pysrc/finetune/__init__.py renamed to llmtune/finetune/__init__.py

@@ -4,20 +4,17 @@
 import csv
 
 from transformers import TextIteratorStreamer
-from rich.console import Console
-from rich.table import Table
-from rich.live import Live
 from rich.text import Text
 from datasets import Dataset
 from transformers import AutoTokenizer, BitsAndBytesConfig
 from peft import AutoPeftModelForCausalLM
 import torch
 
 
-from src.pydantic_models.config_model import Config
-from src.utils.save_utils import DirectoryHelper
-from src.inference.inference import Inference
-from src.ui.rich_ui import RichUI
+from llmtune.pydantic_models.config_model import Config
+from llmtune.utils.save_utils import DirectoryHelper
+from llmtune.inference.generics import Inference
+from llmtune.ui.rich_ui import RichUI
 
 
 # TODO: Add type hints please!