diff --git a/dreamer4/dreamer4.py b/dreamer4/dreamer4.py
index da25586..6130b7a 100644
--- a/dreamer4/dreamer4.py
+++ b/dreamer4/dreamer4.py
@@ -1900,7 +1900,7 @@ class DynamicsWorldModel(Module):
         gae_lambda = 0.95,
         ppo_eps_clip = 0.2,
         pmpo_pos_to_neg_weight = 0.5, # pos and neg equal weight
-        pmpo_kl_div_loss_weight = 1.,
+        pmpo_kl_div_loss_weight = .3,
         value_clip = 0.4,
         policy_entropy_weight = .01,
         gae_use_accelerated = False
diff --git a/pyproject.toml b/pyproject.toml
index 1830d30..8693390 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "dreamer4"
-version = "0.0.99"
+version = "0.0.100"
 description = "Dreamer 4"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }