use uv, update deps

.vscode/extensions.json

@@ -0,0 +1,6 @@
+{
+    "recommendations": [
+        "ms-toolsai.jupyter",
+        "ms-python.python"
+    ]
+}

LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2024 Victor Umesiobi
+Copyright (c) 2025 wassname
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

pyproject.toml

@@ -31,6 +31,10 @@ dependencies = [
     # "lightning>=2.1.0",
     # "torchinfo>=1.8.0",
     # "peft>=0.5.0",
+    "datasets>=3.2.0",
+    "wandb>=0.19.2",
+    "einops>=0.8.0",
+    "evaluate>=0.4.3",
 ]
 
 [project.urls]

requirements.txt

@@ -1 +0,0 @@
-kagglehub

setup.py

@@ -1,8 +0,0 @@
-import kagglehub
-
-kagglehub.login()
-
-# Download latest version
-path = kagglehub.model_download("google/gemma-2/transformers/gemma-2-2b")
-
-print("Path to model files:", path)