Spaces:

jbilcke-hf
/

ai-comic-factory

Running on CPU Upgrade

jbilcke-hf HF staff commited on Sep 12, 2023

Commit

a3bf1f1

•

1 Parent(s): bcd2d46

Update src/app/queries/predict.ts

Files changed (1) hide show

src/app/queries/predict.ts CHANGED Viewed

@@ -55,7 +55,8 @@ export async function predict(inputs: string) {
       parameters: {
         do_sample: true,
         // we don't require a lot of token for our task
-        max_new_tokens: 330, // 1150,
         return_full_text: false,
       }
     })) {

       parameters: {
         do_sample: true,
         // we don't require a lot of token for our task
+        // but to be safe, let's count ~110 tokens per panel
+        max_new_tokens: 450, // 1150,
         return_full_text: false,
       }
     })) {