SDXL-Model-Merger / src /gpu_decorator.py
Kyle Pearson
Add zero-gpu support, enhance model export with quantization/gpu acceleration helpers, optimize inference pipeline with vae fixes, modernize pipeline loading with unified decorators, implement gpu decorator infrastructure.
3631a8e
raw
history blame contribute delete
242 Bytes
"""ZeroGPU compatibility decorator for HuggingFace Spaces."""
try:
import spaces
GPU = spaces.GPU
except ImportError:
def GPU(func=None, duration=None):
if func is None:
return lambda f: f
return func