update the requirement.

Signed-off-by: wxywb <xy.wang@zilliz.com>
2 years ago · 4da37ac8c6
2 changed files with 5 additions and 1 deletions
--- a/clip_caption_reward.py
+++ b/clip_caption_reward.py
@ -15,7 +15,6 @@ import sys
 import pathlib
 import json
 from pathlib import Path
-from PIL import Image
 import numpy as np
 import torch
 from torch import nn
--- a/requirements.txt
+++ b/requirements.txt
@ -0,0 +1,5 @@
+numpy
+torch
+torchvision
+timm
+towhee