Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,8 @@ model = Llama(
|
|
8 |
model_path=hf_hub_download(
|
9 |
repo_id=os.environ.get("REPO_ID", "Lyte/QuadConnect2.5-0.5B-v0.0.4b"),
|
10 |
filename=os.environ.get("MODEL_FILE", "unsloth.Q8_0.gguf"),
|
11 |
-
)
|
|
|
12 |
)
|
13 |
|
14 |
SYSTEM_PROMPT = """You are an expert Connect Four player. The game is played on a 6x7 grid where pieces fall to the lowest available position in each column due to gravity.
|
@@ -237,7 +238,7 @@ def create_interface():
|
|
237 |
],
|
238 |
temperature=0.8,
|
239 |
top_p=0.95,
|
240 |
-
|
241 |
)
|
242 |
|
243 |
ai_response = response['choices'][0]['message']['content']
|
|
|
8 |
model_path=hf_hub_download(
|
9 |
repo_id=os.environ.get("REPO_ID", "Lyte/QuadConnect2.5-0.5B-v0.0.4b"),
|
10 |
filename=os.environ.get("MODEL_FILE", "unsloth.Q8_0.gguf"),
|
11 |
+
),
|
12 |
+
n_ctx=16384
|
13 |
)
|
14 |
|
15 |
SYSTEM_PROMPT = """You are an expert Connect Four player. The game is played on a 6x7 grid where pieces fall to the lowest available position in each column due to gravity.
|
|
|
238 |
],
|
239 |
temperature=0.8,
|
240 |
top_p=0.95,
|
241 |
+
max_tokens=2048
|
242 |
)
|
243 |
|
244 |
ai_response = response['choices'][0]['message']['content']
|