change VPT policy name

CraftJarvis · Dec 13, 2024 · 0e54363 · 0e54363
1 parent 7feb465
commit 0e54363
Show file tree

Hide file tree

Showing 6 changed files with 11 additions and 9 deletions.
diff --git a/.gitignore b/.gitignore
@@ -175,4 +175,6 @@ node_modules/
 record/
 
 minestudio/models/realtime_sam/notebooks/*.jpg
-minestudio/models/realtime_sam/checkpoints/*.pt
+minestudio/models/realtime_sam/checkpoints/*.pt
+
+outputs
diff --git a/minestudio/benchmark/test_pipeline.py b/minestudio/benchmark/test_pipeline.py
@@ -11,7 +11,7 @@
 from minestudio.inference import EpisodePipeline, MineGenerator, InfoBaseFilter
 from minestudio.benchmark.utility.read_conf import convert_yaml_to_callbacks
 from minestudio.benchmark.utility.task_call import TaskCallback
-from minestudio.models import OpenAIPolicy, load_vpt_policy
+from minestudio.models import VPTPolicy, load_vpt_policy
 from minestudio.simulator.callbacks import (
     RecordCallback, 
     RewardsCallback, 

diff --git a/minestudio/inference/example.py b/minestudio/inference/example.py
@@ -7,7 +7,7 @@
 
 from minestudio.simulator import MinecraftSim
 from minestudio.simulator.callbacks import RecordCallback, SpeedTestCallback
-from minestudio.models import OpenAIPolicy, load_vpt_policy
+from minestudio.models import VPTPolicy, load_vpt_policy
 
 if __name__ == '__main__':
 

diff --git a/minestudio/inference/example_online.py b/minestudio/inference/example_online.py
@@ -1,10 +1,10 @@
 from minestudio.simulator import MinecraftSim
 from minestudio.simulator.callbacks import RecordCallback, SpeedTestCallback, SummonMobsCallback, MaskActionsCallback, RewardsCallback, CommandsCallback, JudgeResetCallback, FastResetCallback
-from minestudio.models import OpenAIPolicy, load_openai_policy
+from minestudio.models import VPTPolicy, load_vpt_policy
 
 if __name__ == '__main__':
 
-    policy = load_openai_policy(
+    policy = load_vpt_policy(
         model_path="/nfs-shared/jarvisbase/pretrained/foundation-model-2x.model",
         weights_path="/nfs-shared/jarvisbase/pretrained/rl-from-early-game-2x.weights"
     ).to("cuda")

diff --git a/minestudio/online/rollout/rollout_worker.py b/minestudio/online/rollout/rollout_worker.py
@@ -321,10 +321,10 @@ def progress_handler(self, *,
     #     return sim
 
     # def policy_generator():
-    #     from minestudio.models.openai_vpt.body import load_openai_policy
+    #     from minestudio.models.openai_vpt.body import load_vpt_policy
     #     model_path = '/nfs-shared/jarvisbase/pretrained/foundation-model-2x.model'
     #     weights_path = '/nfs-shared/jarvisbase/pretrained/rl-from-early-game-2x.weights'
-    #     policy = load_openai_policy(model_path, weights_path)
+    #     policy = load_vpt_policy(model_path, weights_path)
     #     return policy
 
     # worker = RolloutWorker(

diff --git a/minestudio/online/run/config.py b/minestudio/online/run/config.py
@@ -132,8 +132,8 @@ def env_generator():
     return sim
 
 def policy_generator():
-    from minestudio.models.openai_vpt.body import load_openai_policy
+    from minestudio.models import load_vpt_policy
     model_path = '/nfs-shared/jarvisbase/pretrained/foundation-model-2x.model'
     weights_path = '/nfs-shared/jarvisbase/pretrained/rl-from-early-game-2x.weights'
-    policy = load_openai_policy(model_path, weights_path)
+    policy = load_vpt_policy(model_path, weights_path)
     return policy