increased timeout (larger models take longer)

This commit is contained in:
Dustin 2024-08-14 22:56:25 -04:00
parent e7ef9efecb
commit 247b5d849d

View File

@ -112,7 +112,7 @@ class ollamarama:
"repeat_penalty": self.repeat_penalty "repeat_penalty": self.repeat_penalty
} }
} }
response = requests.post(self.api_url, json=data, timeout=120) response = requests.post(self.api_url, json=data, timeout=300) #may need to increase for larger models
response.raise_for_status() response.raise_for_status()
data = response.json() data = response.json()