Spaces:

bitwise42
/

scriptify-api

Running

henok3878 commited on Jun 13

Commit

99b28cc

1 Parent(s): b302051

refactor: change sampling to torch.inference_mode

Files changed (1) hide show

main.py CHANGED Viewed

@@ -202,7 +202,7 @@ def generate_strokes(
                                   char_seq_lengths=priming_text_len_tensor)
-    with torch.no_grad():
         try:
             stroke_tensors = SCRIPTED_MODEL.sample(
                 char_seq,

                                   char_seq_lengths=priming_text_len_tensor)
+    with torch.inference_mode():
         try:
             stroke_tensors = SCRIPTED_MODEL.sample(
                 char_seq,