ccm commited on
Commit
fa7c5f2
β€’
1 Parent(s): d7a54c3

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +0 -2
main.py CHANGED
@@ -64,8 +64,6 @@ def search(query: str, k: int) -> tuple[str]:
64
  pipe = transformers.pipeline(
65
  "text-generation",
66
  model="Qwen/Qwen2-0.5B-Instruct",
67
- # model="microsoft/Phi-3-medium-128k-instruct-onnx-cuda",
68
- # model="microsoft/Phi-3-medium-128k-instruct",
69
  trust_remote_code=True,
70
  max_new_tokens = 512,
71
  device="cuda:0",
 
64
  pipe = transformers.pipeline(
65
  "text-generation",
66
  model="Qwen/Qwen2-0.5B-Instruct",
 
 
67
  trust_remote_code=True,
68
  max_new_tokens = 512,
69
  device="cuda:0",