Hugging Face's logo

Spaces:
mridulk
/
phylo-diffusion
Running

App Files Community
1
mridulk commited on
Commit
e380d2b
·
verified ·
1 Parent(s): b8cee61

Update model_files/fishes/2024-03-01T23-15-36-project.yaml

Browse files
Files changed (1) hide show
  1. model_files/fishes/2024-03-01T23-15-36-project.yaml +1 -0
model_files/fishes/2024-03-01T23-15-36-project.yaml CHANGED
@@ -68,6 +68,7 @@ model:
68
  - 38
69
  embed_dim: 512
70
  key: class_to_node
 
71
  data:
72
  target: main.DataModuleFromConfig
73
  params:
 
68
  - 38
69
  embed_dim: 512
70
  key: class_to_node
71
+ device: cpu
72
  data:
73
  target: main.DataModuleFromConfig
74
  params: