Spaces:
Sleeping
Sleeping
import functools | |
def GPU(func): | |
"""Decorator to run a function on the fake GPU | |
to get comparable with HF Space""" | |
# Preserves original function's metadata | |
def wrapper(*args, **kwargs): | |
result = func(*args, **kwargs) | |
return result | |
return wrapper |