File size: 391 Bytes
01bb284 ac2323a c4944bf ac2323a c4944bf |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
# decorators.py
from spaces import GPU # Asegúrate de que 'spaces' está correctamente importado desde Hugging Face
# Decorador personalizado que usa GPU si está disponible
def gpu_decorator(duration=100):
def decorator(func):
@GPU(duration=duration)
def wrapper(*args, **kwargs):
return func(*args, **kwargs)
return wrapper
return decorator
|