Spaces:

Singularity666
/

RadiXGPT_

Runtime error

Singularity666 commited on Jun 19, 2023

Commit

ffaac97

1 Parent(s): 3b95160

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -37,7 +37,7 @@ class CFG:
     pretrained = True  # for both image encoder and text encoder
     trainable = True  # for both image encoder and text encoder
-    temperature = 5.0
     # image size
     size = 224

     pretrained = True  # for both image encoder and text encoder
     trainable = True  # for both image encoder and text encoder
+    temperature = 1.0
     # image size
     size = 224