1inkusFace commited on
Commit
14f5ced
·
verified ·
1 Parent(s): be8af01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -48,7 +48,7 @@ FTP_PASS = os.getenv("FTP_PASS")
48
  FTP_DIR = "1ink.us/stable_diff/" # Remote directory on FTP server
49
 
50
  DESCRIPTIONXX = """
51
- ## ⚡⚡⚡⚡ REALVISXL V5.0 BF16 IP Adapter ⚡⚡⚡⚡
52
  """
53
 
54
  examples = [
@@ -282,8 +282,8 @@ def captioning(img):
282
  **inputs,
283
  do_sample=False,
284
  num_beams=5,
285
- max_length=96,
286
- #min_length=1,
287
  top_p=0.9,
288
  repetition_penalty=1.5,
289
  length_penalty=1.0,
@@ -296,7 +296,7 @@ def captioning(img):
296
 
297
  # Continue conversation:
298
  inputf = processor5(images=img, text=generated_text + 'So therefore', return_tensors="pt").to('cuda')
299
- generated_ids = model5.generate(**inputf, max_length=96)
300
  generated_text = processor5.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
301
  response_text = generated_text.replace(generated_text, "").strip()
302
  print(response_text)
 
48
  FTP_DIR = "1ink.us/stable_diff/" # Remote directory on FTP server
49
 
50
  DESCRIPTIONXX = """
51
+ ## ⚡⚡⚡⚡ REALVISXL V5.0 BF16 IP Adapter Test B ⚡⚡⚡⚡
52
  """
53
 
54
  examples = [
 
282
  **inputs,
283
  do_sample=False,
284
  num_beams=5,
285
+ max_length=128,
286
+ min_length=42,
287
  top_p=0.9,
288
  repetition_penalty=1.5,
289
  length_penalty=1.0,
 
296
 
297
  # Continue conversation:
298
  inputf = processor5(images=img, text=generated_text + 'So therefore', return_tensors="pt").to('cuda')
299
+ generated_ids = model5.generate(**inputf, max_length=192)
300
  generated_text = processor5.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
301
  response_text = generated_text.replace(generated_text, "").strip()
302
  print(response_text)