Lyte commited on
Commit
2547dd4
·
verified ·
1 Parent(s): d2c0f15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- #max_tokens=4096
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']