bedead commited on
Commit
7d448bb
1 Parent(s): 5e010ad

Update config/inference_config.yaml

Browse files
Files changed (1) hide show
  1. config/inference_config.yaml +1 -1
config/inference_config.yaml CHANGED
@@ -1,4 +1,4 @@
1
- device: "cuda"
2
  root_path: assets/gradio_examples
3
  image_encoder_path: image_encoder
4
  alpha_clip_ckpt_pth: clip_l14_grit+mim_fultune_6xe.pth
 
1
+ device: "cpu"
2
  root_path: assets/gradio_examples
3
  image_encoder_path: image_encoder
4
  alpha_clip_ckpt_pth: clip_l14_grit+mim_fultune_6xe.pth