remove old models

alex-petrenko · Jan 24, 2025 · d284101 · d284101
1 parent 8237d66
commit d284101
Show file tree

Hide file tree

Showing 6 changed files with 2 additions and 972 deletions.
diff --git a/sf_examples/nethack/models/__init__.py b/sf_examples/nethack/models/__init__.py
@@ -1,11 +1,4 @@
-from sf_examples.nethack.models.chaotic_dwarf import ChaoticDwarvenGPT5
-from sf_examples.nethack.models.scaled import ScaledNet
-from sf_examples.nethack.models.simba import SimbaActorEncoder, SimbaCriticEncoder
+from sf_examples.nethack.models.vit import ViTActorEncoder, ViTCriticEncoder
 
-MODELS = [
-    ChaoticDwarvenGPT5,
-    ScaledNet,
-    SimbaActorEncoder,
-    SimbaCriticEncoder,
-]
+MODELS = [ViTActorEncoder, ViTCriticEncoder]
 MODELS_LOOKUP = {c.__name__: c for c in MODELS}
diff --git a/sf_examples/nethack/models/chaotic_dwarf.py b/sf_examples/nethack/models/chaotic_dwarf.py