yyk19 commited on
Commit
9c6e0e6
·
1 Parent(s): ee736c7

support cpu version of CLIP text encoder

Browse files
Files changed (1) hide show
  1. config.yaml +1 -0
config.yaml CHANGED
@@ -85,3 +85,4 @@ model:
85
  params:
86
  freeze: True
87
  layer: "penultimate"
 
 
85
  params:
86
  freeze: True
87
  layer: "penultimate"
88
+ device: "cpu"