Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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=
|
286 |
-
|
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=
|
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)
|