Spaces:
Sleeping
Sleeping
Commit
·
b50ad21
1
Parent(s):
add2495
changed the filerserver
Browse files
app.py
CHANGED
@@ -64,8 +64,8 @@ config, base_model, tokenizer = build_or_load_gen_model(args)
|
|
64 |
print("Base model architecture and tokenizer loaded.")
|
65 |
|
66 |
# Download the fine-tuned weights via ClearML using your injected creds
|
67 |
-
|
68 |
-
finetuned_weights_path =
|
69 |
adapter_dir = os.path.dirname(finetuned_weights_path)
|
70 |
print(f"Fine-tuned adapter weights downloaded to directory: {adapter_dir}")
|
71 |
|
|
|
64 |
print("Base model architecture and tokenizer loaded.")
|
65 |
|
66 |
# Download the fine-tuned weights via ClearML using your injected creds
|
67 |
+
task = Task.get_task(task_id="9cc33fb4d1d54378b691188c5e230253")
|
68 |
+
finetuned_weights_path = task.artifacts["lora-pytorch-bin"].get_local_copy()
|
69 |
adapter_dir = os.path.dirname(finetuned_weights_path)
|
70 |
print(f"Fine-tuned adapter weights downloaded to directory: {adapter_dir}")
|
71 |
|
test.py
ADDED
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from clearml import Task
|
2 |
+
task = Task.get_task(task_id="9cc33fb4d1d54378b691188c5e230253")
|
3 |
+
print("Artifacts keys:", list(task.artifacts.keys()))
|
4 |
+
# Download to verify contents:
|
5 |
+
path = task.artifacts["lora-pytorch-bin"].get_local_copy()
|
6 |
+
print("Downloaded to:", path)
|