arnavmishra4 commited on
Commit
6ea4e9b
·
verified ·
1 Parent(s): 3978d87

Update models/m5_cfdna/loader.py

Browse files
Files changed (1) hide show
  1. models/m5_cfdna/loader.py +1 -1
models/m5_cfdna/loader.py CHANGED
@@ -8,7 +8,7 @@ Usage:
8
 
9
  import torch
10
  from pathlib import Path
11
- from model import Pleiades
12
 
13
  # Keys the training code may have used to store the model state dict
14
  _MODEL_STATE_KEYS = ("model", "model_state_dict", "state_dict")
 
8
 
9
  import torch
10
  from pathlib import Path
11
+ from models.model import Pleiades
12
 
13
  # Keys the training code may have used to store the model state dict
14
  _MODEL_STATE_KEYS = ("model", "model_state_dict", "state_dict")