Spaces:
Sleeping
Sleeping
Commit
·
b6b4764
1
Parent(s):
ad11fe1
chore: made optims
Browse filesSigned-off-by: Suvaditya Mukherjee <[email protected]>
app.py
CHANGED
@@ -125,10 +125,22 @@ if __name__ == "__main__":
|
|
125 |
parallel=0,
|
126 |
)
|
127 |
|
128 |
-
wandb.
|
|
|
|
|
129 |
|
130 |
model_name = "Qwen/Qwen2.5-3B-Instruct"
|
131 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
132 |
@spaces.GPU
|
133 |
def rag_process(message, chat_history):
|
134 |
# Append current user message to chat history
|
|
|
125 |
parallel=0,
|
126 |
)
|
127 |
|
128 |
+
wandb.login(
|
129 |
+
api_key=os.getenv("WANDB_API_KEY")
|
130 |
+
)
|
131 |
|
132 |
model_name = "Qwen/Qwen2.5-3B-Instruct"
|
133 |
|
134 |
+
wandb.init(
|
135 |
+
project="resume-rag",
|
136 |
+
name="zerogpu-run",
|
137 |
+
save_code=True,
|
138 |
+
config={
|
139 |
+
"model_name": model_name,
|
140 |
+
"resume_url": RESUME_URL
|
141 |
+
}
|
142 |
+
)
|
143 |
+
|
144 |
@spaces.GPU
|
145 |
def rag_process(message, chat_history):
|
146 |
# Append current user message to chat history
|