lindsay-qu commited on
Commit
bee9e0d
·
verified ·
1 Parent(s): 2085fdc

Update core/refiner/simple_refiner.py

Browse files
Files changed (1) hide show
  1. core/refiner/simple_refiner.py +1 -5
core/refiner/simple_refiner.py CHANGED
@@ -23,11 +23,7 @@ class SimpleRefiner(BaseRefiner):
23
  image_paths = [image_paths]
24
  for image_path in image_paths:
25
  user_context.append({"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encode_image(image_path.name)}"}})
26
- context = [{"role": "system", "content": self.sys_prompt}] + memory + [{"role": "user", "content": [
27
- {"type": "text", "text": f"{message}"},
28
- ]}]
29
- else:
30
- context = [{"role": "system", "content": self.sys_prompt}] + memory + user_context
31
 
32
  respond_task = asyncio.create_task(self.model.respond_async(context))
33
  await respond_task
 
23
  image_paths = [image_paths]
24
  for image_path in image_paths:
25
  user_context.append({"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encode_image(image_path.name)}"}})
26
+ context = [{"role": "system", "content": self.sys_prompt}] + memory + user_context
 
 
 
 
27
 
28
  respond_task = asyncio.create_task(self.model.respond_async(context))
29
  await respond_task