WalkingOnSaturn commited on
Commit
5de647f
·
verified ·
1 Parent(s): 1bd49e1

fix: pass device as string (Hydra/OmegaConf rejects torch.device)

Browse files
Files changed (1) hide show
  1. server.py +3 -1
server.py CHANGED
@@ -45,7 +45,9 @@ def _load_model():
45
  print("[server] loading Kimodo-SOMA-RP-v1.1 ...", file=sys.stderr, flush=True)
46
  from kimodo import load_model
47
 
48
- _device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 
49
  print(f"[server] device = {_device}", file=sys.stderr, flush=True)
50
 
51
  model, resolved = load_model(
 
45
  print("[server] loading Kimodo-SOMA-RP-v1.1 ...", file=sys.stderr, flush=True)
46
  from kimodo import load_model
47
 
48
+ # Must be a string (kimodo passes this through Hydra/OmegaConf which
49
+ # rejects non-primitive types like torch.device).
50
+ _device = "cuda:0" if torch.cuda.is_available() else "cpu"
51
  print(f"[server] device = {_device}", file=sys.stderr, flush=True)
52
 
53
  model, resolved = load_model(