revert open clip

Author: lshqqytiger

modules/launch_utils.py

@@ -527,7 +527,6 @@ def prepare_environment():
         amd_gpus = []
         try:
             amd_gpus = rocm.get_agents()
-            print('ROCm: AMD toolkit detected')
         except Exception as e:
             print(f'ROCm agent enumerator failed: {e}')
 
@@ -536,6 +535,7 @@ def prepare_environment():
                 print('No ROCm agent was found. Please make sure that graphics driver is installed and up to date.')
             backend = "cpu"
         else:
+            print('ROCm: AMD toolkit detected')
             print(f'ROCm: agents={[gpu.name for gpu in amd_gpus]}')
             if args.device_id is None:
                 index = 0

requirements_versions.txt

@@ -17,13 +17,13 @@ kornia==0.6.7
 lark==1.1.2
 numpy==1.26.4
 omegaconf==2.2.3
-open-clip-torch==3.2.0
+open-clip-torch==2.20.0
 onnx==1.16.2
 onnxruntime
 optimum==1.25.3
 olive-ai
 piexif==1.1.3
-protobuf==4.25.3
+protobuf==3.20.2
 psutil==5.9.5
 pytorch_lightning==1.9.4
 resize-right==0.0.2