shekkari21 commited on
Commit
b50ad21
·
1 Parent(s): add2495

changed the filerserver

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. test.py +6 -0
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
- model_obj = Model(model_id="34e25deb24c64b74b29c8519ed15fe3e")
68
- finetuned_weights_path = model_obj.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
 
 
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)