Spaces:
Sleeping
Sleeping
Factor Studios
commited on
Update http_storage.py
Browse files- http_storage.py +1 -6
http_storage.py
CHANGED
|
@@ -337,7 +337,6 @@ class HTTPGPUStorage:
|
|
| 337 |
def is_model_loaded(self, model_name: str) -> bool:
|
| 338 |
"""Check if a model is loaded via HTTP API"""
|
| 339 |
try:
|
| 340 |
-
import urllib.parse
|
| 341 |
response = self._make_request(
|
| 342 |
"GET",
|
| 343 |
f"/models/{model_name}/status",
|
|
@@ -358,10 +357,6 @@ class HTTPGPUStorage:
|
|
| 358 |
retry_delay = 2
|
| 359 |
last_error = None
|
| 360 |
|
| 361 |
-
# URL encode model name for safety
|
| 362 |
-
import urllib.parse
|
| 363 |
-
encoded_name = urllib.parse.quote(model_name, safe='')
|
| 364 |
-
|
| 365 |
for attempt in range(max_retries):
|
| 366 |
try:
|
| 367 |
# Ensure connection is active and create new session if needed
|
|
@@ -388,7 +383,7 @@ class HTTPGPUStorage:
|
|
| 388 |
|
| 389 |
response = self._make_request(
|
| 390 |
'POST',
|
| 391 |
-
f'/models/{
|
| 392 |
json=request_data,
|
| 393 |
timeout=300 # 5 minutes timeout for model loading
|
| 394 |
)
|
|
|
|
| 337 |
def is_model_loaded(self, model_name: str) -> bool:
|
| 338 |
"""Check if a model is loaded via HTTP API"""
|
| 339 |
try:
|
|
|
|
| 340 |
response = self._make_request(
|
| 341 |
"GET",
|
| 342 |
f"/models/{model_name}/status",
|
|
|
|
| 357 |
retry_delay = 2
|
| 358 |
last_error = None
|
| 359 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 360 |
for attempt in range(max_retries):
|
| 361 |
try:
|
| 362 |
# Ensure connection is active and create new session if needed
|
|
|
|
| 383 |
|
| 384 |
response = self._make_request(
|
| 385 |
'POST',
|
| 386 |
+
f'/models/{model_name}/load',
|
| 387 |
json=request_data,
|
| 388 |
timeout=300 # 5 minutes timeout for model loading
|
| 389 |
)
|