vdblm commited on
Commit
ccfc508
·
verified ·
1 Parent(s): 25d06e7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -7
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- license: other
3
  tags:
4
  - Causal-Inference
5
  - CausalPFN
@@ -17,12 +17,13 @@ The model is based on a transformer architecture with uncertainty quantification
17
 
18
  ## Requirements
19
 
20
- - Python ≥ 3.10
21
- - torch ≥ 2.0
22
- - numpy ≥ 1.24
 
 
23
  - tqdm
24
- - faiss-cpu ≥ 1.9.0
25
- - scikit-learn ≥ 1.5.2
26
  - huggingface_hub
27
 
28
  ## Installation
@@ -84,4 +85,4 @@ If you use this model in your research, please cite:
84
 
85
  ## License
86
 
87
- This model is licensed under a custom license. See the [LICENSE](https://github.com/vdblm/CausalPFN/blob/main/LICENSE) file for details.
 
1
  ---
2
+ license: apache-2.0
3
  tags:
4
  - Causal-Inference
5
  - CausalPFN
 
17
 
18
  ## Requirements
19
 
20
+
21
+ - Python 3.10+
22
+ - PyTorch 2.3+
23
+ - NumPy
24
+ - scikit-learn
25
  - tqdm
26
+ - faiss-cpu
 
27
  - huggingface_hub
28
 
29
  ## Installation
 
85
 
86
  ## License
87
 
88
+ This model is licensed under [Apache-2.0](https://github.com/vdblm/CausalPFN/blob/main/LICENSE).