HuggingFace ZERO Spaceでローカルでもimport spacesと共存する

ローカルだと import spaces でエラーが出ることがある(Windowsだけかバージョンによるかも)ので、いちいちコメントアウトすることになってちょっと面倒です。

なので、インポートできなかった場合に spaces という名前のクラスと GPU のデコレータを作ってあげることで、ローカルでもコメントアウトせずに Spaces と同じコードが使えます。(もうちょい簡単に書けるかも？)

try:
    import spaces
except:
    class spaces:
        @staticmethod
        def GPU(duration: int):
            return lambda x: x # ローカルでは特になんもしない

@spaces.GPU(duration=45)
def generate():
    pass

Ubuntu環境で実行したら特になんもエラー出なかったので非常に謎です