Spaces:
Running
on
Zero
Running
on
Zero
AdrienB134
commited on
Commit
•
e56b146
1
Parent(s):
b8734cc
fearf
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
|
|
28 |
|
29 |
@spaces.GPU
|
30 |
def model_inference(
|
31 |
-
images, text, assistant_prefix=
|
32 |
repetition_penalty=1.2, top_p=0.8
|
33 |
):
|
34 |
## Load idefics
|
|
|
28 |
|
29 |
@spaces.GPU
|
30 |
def model_inference(
|
31 |
+
images, text, assistant_prefix= "Répond en faisant de belles phrases", decoding_strategy = "Greedy", temperature= 0.4, max_new_tokens=512,
|
32 |
repetition_penalty=1.2, top_p=0.8
|
33 |
):
|
34 |
## Load idefics
|