Spaces:

fffiloni
/

RB-Modulation

Sleeping

fffiloni commited on Sep 2, 2024

Commit

e936216

verified ·

1 Parent(s): 6097263

fix typo

Files changed (1) hide show

app.py CHANGED Viewed

@@ -129,7 +129,7 @@ models_rbm.generator.eval().requires_grad_(False)
 def infer(ref_style_file, style_description, caption, progress):
-    global models_rbm, models_b, device:
     if low_vram:
         models_to(models_rbm, device=device, excepts=["generator", "previewer"])

 def infer(ref_style_file, style_description, caption, progress):
+    global models_rbm, models_b, device
     if low_vram:
         models_to(models_rbm, device=device, excepts=["generator", "previewer"])