ai-georeferencer / spaces_optimizer.py
ziffir's picture
Create spaces_optimizer.py
5173b8d verified
raw
history blame
743 Bytes
# spaces_optimizer.py - Spaces optimizasyonu
import gc
import psutil
import threading
from functools import lru_cache
class SpacesOptimizer:
"""Hugging Face Spaces için optimizasyon sınıfı"""
def __init__(self):
self.memory_limit = 16 * 1024 * 1024 * 1024 # 16GB
self.optimize_memory()
def optimize_memory(self):
"""Bellek optimizasyonu"""
# Garbage collection
gc.collect()
if torch.cuda.is_available():
torch.cuda.empty_cache()
# Thread sayısını optimize et
torch.set_num_threads(2)
@lru_cache(maxsize=1)
def load_model_cached(self):
"""Modeli cache'le"""
return ProfessionalGeoReferencingSystem()