Update README.md
Browse files
README.md
CHANGED
|
@@ -23,6 +23,13 @@ PreceptCLIP-Emotions is a model designed to predict the emotions that an image e
|
|
| 23 |
- *Learning Rate*: 0.0001
|
| 24 |
- *Batch Size*: 32
|
| 25 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
## Usage
|
| 27 |
|
| 28 |
To use the model for inference:
|
|
|
|
| 23 |
- *Learning Rate*: 0.0001
|
| 24 |
- *Batch Size*: 32
|
| 25 |
|
| 26 |
+
## requirements
|
| 27 |
+
|
| 28 |
+
cudatoolkit=11.7
|
| 29 |
+
torchvision=0.14.0
|
| 30 |
+
transformers==4.45.2
|
| 31 |
+
peft==0.14.0
|
| 32 |
+
|
| 33 |
## Usage
|
| 34 |
|
| 35 |
To use the model for inference:
|