Skip to content

Commit 4ff8b4e

Browse files
authored
📜 Convert set to list of tags (#4092)
1 parent 6356343 commit 4ff8b4e

14 files changed

+14
-14
lines changed

trl/trainer/bco_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1553,7 +1553,7 @@ def create_model_card(
15531553
model_name=model_name,
15541554
hub_model_id=self.hub_model_id,
15551555
dataset_name=dataset_name,
1556-
tags=tags,
1556+
tags=list(tags),
15571557
wandb_url=wandb.run.url if is_wandb_available() and wandb.run is not None else None,
15581558
comet_url=get_comet_experiment_url(),
15591559
trainer_name="BCO",

trl/trainer/cpo_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1127,7 +1127,7 @@ def create_model_card(
11271127
model_name=model_name,
11281128
hub_model_id=self.hub_model_id,
11291129
dataset_name=dataset_name,
1130-
tags=tags,
1130+
tags=list(tags),
11311131
wandb_url=wandb.run.url if is_wandb_available() and wandb.run is not None else None,
11321132
comet_url=get_comet_experiment_url(),
11331133
trainer_name="CPO",

trl/trainer/dpo_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2008,7 +2008,7 @@ def create_model_card(
20082008
model_name=model_name,
20092009
hub_model_id=self.hub_model_id,
20102010
dataset_name=dataset_name,
2011-
tags=tags,
2011+
tags=list(tags),
20122012
wandb_url=wandb.run.url if is_wandb_available() and wandb.run is not None else None,
20132013
comet_url=get_comet_experiment_url(),
20142014
trainer_name="DPO",

trl/trainer/gkd_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -482,7 +482,7 @@ def create_model_card(
482482
model_name=model_name,
483483
hub_model_id=self.hub_model_id,
484484
dataset_name=dataset_name,
485-
tags=tags,
485+
tags=list(tags),
486486
wandb_url=wandb.run.url if is_wandb_available() and wandb.run is not None else None,
487487
comet_url=get_comet_experiment_url(),
488488
trainer_name="GKD",

trl/trainer/grpo_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1890,7 +1890,7 @@ def create_model_card(
18901890
model_name=model_name,
18911891
hub_model_id=self.hub_model_id,
18921892
dataset_name=dataset_name,
1893-
tags=tags,
1893+
tags=list(tags),
18941894
wandb_url=wandb.run.url if is_wandb_available() and wandb.run is not None else None,
18951895
comet_url=get_comet_experiment_url(),
18961896
trainer_name="GRPO",

trl/trainer/kto_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1733,7 +1733,7 @@ def create_model_card(
17331733
model_name=model_name,
17341734
hub_model_id=self.hub_model_id,
17351735
dataset_name=dataset_name,
1736-
tags=tags,
1736+
tags=list(tags),
17371737
wandb_url=wandb.run.url if is_wandb_available() and wandb.run is not None else None,
17381738
comet_url=get_comet_experiment_url(),
17391739
trainer_name="KTO",

trl/trainer/nash_md_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -554,7 +554,7 @@ def create_model_card(
554554
model_name=model_name,
555555
hub_model_id=self.hub_model_id,
556556
dataset_name=dataset_name,
557-
tags=tags,
557+
tags=list(tags),
558558
wandb_url=wandb.run.url if is_wandb_available() and wandb.run is not None else None,
559559
comet_url=get_comet_experiment_url(),
560560
trainer_name="Nash-MD",

trl/trainer/online_dpo_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1565,7 +1565,7 @@ def create_model_card(
15651565
model_name=model_name,
15661566
hub_model_id=self.hub_model_id,
15671567
dataset_name=dataset_name,
1568-
tags=tags,
1568+
tags=list(tags),
15691569
wandb_url=wandb.run.url if is_wandb_available() and wandb.run is not None else None,
15701570
comet_url=get_comet_experiment_url(),
15711571
trainer_name="Online DPO",

trl/trainer/orpo_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1087,7 +1087,7 @@ def create_model_card(
10871087
model_name=model_name,
10881088
hub_model_id=self.hub_model_id,
10891089
dataset_name=dataset_name,
1090-
tags=tags,
1090+
tags=list(tags),
10911091
wandb_url=wandb.run.url if is_wandb_available() and wandb.run is not None else None,
10921092
comet_url=get_comet_experiment_url(),
10931093
trainer_name="ORPO",

trl/trainer/ppo_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -849,7 +849,7 @@ def create_model_card(
849849
model_name=model_name,
850850
hub_model_id=self.hub_model_id,
851851
dataset_name=dataset_name,
852-
tags=tags,
852+
tags=list(tags),
853853
wandb_url=wandb.run.url if is_wandb_available() and wandb.run is not None else None,
854854
comet_url=get_comet_experiment_url(),
855855
trainer_name="PPO",

0 commit comments

Comments
 (0)