x-evolution 0.1.24__tar.gz → 0.1.25__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: x-evolution
3
- Version: 0.1.24
3
+ Version: 0.1.25
4
4
  Summary: x-evolution
5
5
  Project-URL: Homepage, https://pypi.org/project/x-evolution/
6
6
  Project-URL: Repository, https://github.com/lucidrains/x-evolution
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "x-evolution"
3
- version = "0.1.24"
3
+ version = "0.1.25"
4
4
  description = "x-evolution"
5
5
  authors = [
6
6
  { name = "Phil Wang", email = "lucidrains@gmail.com" }
@@ -7,10 +7,10 @@
7
7
  # ]
8
8
  # ///
9
9
 
10
- import os
11
- os.environ["NCCL_P2P_DISABLE"] = "1"
12
- os.environ["NCCL_IB_DISABLE"] = "1"
13
- os.environ["MUJOCO_GL"] = "osmesa"
10
+ # import os
11
+ # os.environ["NCCL_P2P_DISABLE"] = "1"
12
+ # os.environ["NCCL_IB_DISABLE"] = "1"
13
+ # os.environ["MUJOCO_GL"] = "osmesa"
14
14
 
15
15
  from shutil import rmtree
16
16
  import gymnasium as gym
@@ -78,7 +78,7 @@ class HumanoidEnvironment(Module):
78
78
 
79
79
  # sample and then bound and scale to -0.4 to 0.4
80
80
 
81
- std = softclamp((0.5 * log_var).exp(), 10.)
81
+ std = (0.5 * softclamp(log_var, 5.)).exp()
82
82
  sampled = mean + torch.randn_like(mean) * std
83
83
  action = sampled.tanh() * 0.4
84
84
 
File without changes
File without changes
File without changes
File without changes