RealVis_v5.0_BF16_IP_B

Running on Zero

1inkusFace commited on Jan 22

Commit

c12d5e6

verified ·

1 Parent(s): 4f48370

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,
@@ -295,7 +295,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,
@@ -315,7 +315,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,
+        text_decoding_method = "Nucleus sampling",
         num_beams=1,
         max_length=128,
         min_length=64,
         #with torch.no_grad():
         generated_ids = model5.generate(
             **inputs,
+            text_decoding_method = "Nucleus sampling",
             num_beams=1,
             max_length=64,
             #min_length=16,
     ).to('cuda')
     generated_ids = model5.generate(
             **inputf,
+            text_decoding_method = "Nucleus sampling",
             num_beams=1,
             max_length=96,
             min_length=64,