RealVis_v5.0_BF16_IP_B

Running on Zero

1inkusFace commited on 17 days ago

Commit

837ba80

verified ·

1 Parent(s): fd60bf9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -275,7 +275,7 @@ def captioning(img):
     '''
     generated_ids = model5.generate(
         **inputsa,
-        text_decoding_method = "Nucleus sampling",
         num_beams=1,
         max_length=128,
         min_length=64,
@@ -296,7 +296,7 @@ def captioning(img):
         #with torch.no_grad():
         generated_ids = model5.generate(
             **inputs,
-            text_decoding_method = "Nucleus sampling",
             num_beams=1,
             max_length=64,
             #min_length=16,
@@ -316,7 +316,7 @@ def captioning(img):
     ).to('cuda')
     generated_ids = model5.generate(
             **inputf,
-            text_decoding_method = "Nucleus sampling",
             num_beams=1,
             max_length=96,
             min_length=64,

     '''
     generated_ids = model5.generate(
         **inputsa,
+        do_sample=False,
         num_beams=1,
         max_length=128,
         min_length=64,
         #with torch.no_grad():
         generated_ids = model5.generate(
             **inputs,
+            do_sample=False,
             num_beams=1,
             max_length=64,
             #min_length=16,
     ).to('cuda')
     generated_ids = model5.generate(
             **inputf,
+            do_sample=False,
             num_beams=1,
             max_length=96,
             min_length=64,