Spaces:

thecollabagepatch
/

magenta-retry

Running

thecollabagepatch commited on Aug 19

Commit

207a138

1 Parent(s): e7bec6e

hmm large

Files changed (1) hide show

app.py CHANGED Viewed

@@ -355,7 +355,7 @@ def get_mrt():
     if _MRT is None:
         with _MRT_LOCK:
             if _MRT is None:
-                _MRT = system.MagentaRT(tag="base", guidance_weight=1.0, device="gpu", lazy=False)
     return _MRT
 @app.post("/generate")

     if _MRT is None:
         with _MRT_LOCK:
             if _MRT is None:
+                _MRT = system.MagentaRT(tag="large", guidance_weight=1.0, device="gpu", lazy=False)
     return _MRT
 @app.post("/generate")