Fix gen_in device logic in generate()

2021-11-17 16:37:37 -05:00 · 2021-11-17 16:37:37 -05:00 · 2f0b673b28
parent 485034b6bb
commit 2f0b673b28
1 changed files with 0 additions and 2 deletions
--- a/aiserver.py
+++ b/aiserver.py
@ -1452,8 +1452,6 @@ def generate(txt, min, max, found_entries=set()):
            gen_in = gen_in.to(0)
        elif(vars.hascuda and vars.breakmodel):
            gen_in = gen_in.to(breakmodel.primary_device)
-        elif(vars.hascuda):
-            gen_in = gen_in.to(0)
        else:
            gen_in = gen_in.to('cpu')