def __init__()

in distilvit/curate.py [0:0]

• 14 lines of code
• 3 McCabe index (conditional complexity)

    def __init__(self, args, model_name=MODEL_NAME):
        if torch.cuda.is_available():
            device = torch.device("cuda")
            print("Using CUDA (Nvidia GPU).")
        elif torch.backends.mps.is_available():
            device = torch.device("mps")
            print("Using MPS (Apple Silicon GPU).")
        else:
            device = torch.device("cpu")
            print("Using CPU.")

        self.device = device
        self.model = None
        self.model_name = model_name
        self.args = args