Spaces:

tohoku-nlp
/

Sketch2Diagram

Runtime error

DaddyDaniel commited on Jul 28

Commit

536c9c2

verified ·

1 Parent(s): 5e8ed1e

Update qwen2_inference.py

Files changed (1) hide show

qwen2_inference.py CHANGED Viewed

@@ -30,7 +30,6 @@ def get_model(model_path):
                 attn_implementation="flash_attention_2",
                 token=HUGGINGFACE_TOKEN,
             )
-            model_import = model_import.to("cuda")
             size = {
                 "shortest_edge": 224,
                 "longest_edge": 1024,

                 attn_implementation="flash_attention_2",
                 token=HUGGINGFACE_TOKEN,
             )
             size = {
                 "shortest_edge": 224,
                 "longest_edge": 1024,