lukeingawesome commited on
Commit
1cc6474
·
verified ·
1 Parent(s): 71d3180

Upload LLM2Vec4CXR fine-tuned model

Browse files
Files changed (3) hide show
  1. README.md +7 -2
  2. requirements.txt +5 -1
  3. usage_example.py +11 -0
README.md CHANGED
@@ -50,8 +50,13 @@ LLM2Vec4CXR is a bidirectional language model that converts the base decoder-onl
50
  ### Installation
51
 
52
  ```bash
53
- pip install torch transformers
54
- # Also requires the LLM2Vec wrapper - see the original repository for installation
 
 
 
 
 
55
  ```
56
 
57
  ### Basic Usage
 
50
  ### Installation
51
 
52
  ```bash
53
+ # Install the LLM2Vec4CXR package directly from GitHub
54
+ pip install git+https://github.com/lukeingawesome/llm2vec4cxr.git
55
+
56
+ # Or clone and install in development mode
57
+ git clone https://github.com/lukeingawesome/llm2vec4cxr.git
58
+ cd llm2vec4cxr
59
+ pip install -e .
60
  ```
61
 
62
  ### Basic Usage
requirements.txt CHANGED
@@ -1,4 +1,8 @@
1
- torch>=2.0.0
 
 
 
 
2
  transformers>=4.44.0
3
  accelerate>=0.20.0
4
  flash-attn>=2.5.0
 
1
+ # This model requires the LLM2Vec4CXR package
2
+ # Install with: pip install git+https://github.com/lukeingawesome/llm2vec4cxr.git
3
+
4
+ # Core dependencies (automatically installed with the package)
5
+ torch>=2.4.1
6
  transformers>=4.44.0
7
  accelerate>=0.20.0
8
  flash-attn>=2.5.0
usage_example.py CHANGED
@@ -1,6 +1,17 @@
1
  """
2
  Example usage script for LLM2Vec4CXR model.
3
  This demonstrates how to load and use the model for chest X-ray report analysis.
 
 
 
 
 
 
 
 
 
 
 
4
  """
5
 
6
  import torch
 
1
  """
2
  Example usage script for LLM2Vec4CXR model.
3
  This demonstrates how to load and use the model for chest X-ray report analysis.
4
+
5
+ Prerequisites:
6
+ 1. Install the LLM2Vec4CXR package:
7
+ pip install git+https://github.com/lukeingawesome/llm2vec4cxr.git
8
+
9
+ Or clone and install in development mode:
10
+ git clone https://github.com/lukeingawesome/llm2vec4cxr.git
11
+ cd llm2vec4cxr
12
+ pip install -e .
13
+
14
+ 2. The model will be automatically downloaded from Hugging Face when first used.
15
  """
16
 
17
  import torch