01bb284 c4944bf
1
2
3
4
5
6
7
8
9
10
11
12
13
# decorators.py import spaces # Decorador personalizado que usa GPU si está disponible def gpu_decorator(duration=100): def decorator(func): @spaces.GPU(duration=duration) def wrapper(*args, **kwargs): return func(*args, **kwargs) return wrapper return decorator
