fix: changed sample to True to generate caption using blip model

1 year ago · 94728d978e
parent 95d3d08d27
commit 94728d978e
1 changed files with 1 additions and 1 deletions
--- a/imaginairy/enhancers/describe_image_blip.py
+++ b/imaginairy/enhancers/describe_image_blip.py
@ -57,6 +57,6 @@ def generate_caption(image, min_length=30):

    with torch.no_grad():
        caption = blip_model().generate(
-            gpu_image, sample=False, num_beams=3, max_length=80, min_length=min_length
+            gpu_image, sample=True, num_beams=3, max_length=80, min_length=min_length
        )
    return caption[0]