HemanM commited on
Commit
dfcf870
·
verified ·
1 Parent(s): 1c8999c

Update evo_architecture.py

Browse files
Files changed (1) hide show
  1. evo_architecture.py +1 -8
evo_architecture.py CHANGED
@@ -3,7 +3,6 @@ import json
3
  import os
4
  import uuid
5
  import csv
6
- import torch
7
  import torch.nn as nn
8
 
9
  GENOME_LOG = "genome_log.csv"
@@ -64,12 +63,6 @@ def load_best_genome():
64
  return json.load(f)
65
  return default_config()
66
 
67
- # ✅ FIXED: This now exists and solves your import error
68
  def build_model_from_config(config):
69
  from evo_model import EvoTransformerV22
70
- return EvoTransformerV22(
71
- num_layers=config["num_layers"],
72
- ffn_dim=config["ffn_dim"],
73
- num_heads=config["num_heads"],
74
- memory_enabled=config["memory_enabled"]
75
- )
 
3
  import os
4
  import uuid
5
  import csv
 
6
  import torch.nn as nn
7
 
8
  GENOME_LOG = "genome_log.csv"
 
63
  return json.load(f)
64
  return default_config()
65
 
 
66
  def build_model_from_config(config):
67
  from evo_model import EvoTransformerV22
68
+ return EvoTransformerV22(config)