AlexHung29629
/

test_mllama_v12

Feature Extraction

Model card Files Files and versions

AlexHung29629 commited on Nov 19, 2024

Commit

35d32dc

·

verified ·

1 Parent(s): 57cd967

Update ultravox_config.py

Files changed (1) hide show

ultravox_config.py +1 -1

ultravox_config.py CHANGED Viewed

@@ -28,7 +28,7 @@ class LossFunction(str, Enum):
 @dataclasses.dataclass
 class LossConfig:
-    loss_function: LossFunction = LossFunction.KL_Divergence
     kl_temperature: float = 2.0
     @property

 @dataclasses.dataclass
 class LossConfig:
+    loss_function: LossFunction = LossFunction.CrossEntropy
     kl_temperature: float = 2.0
     @property