geshang commited on
Commit
d855e45
·
verified ·
1 Parent(s): 4a728a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,7 +72,7 @@ try:
72
  MODEL_PATH,
73
  torch_dtype=torch.bfloat16 if DEVICE == "cuda" else torch.float32,
74
  device_map="auto" if DEVICE == "cuda" else None
75
- ).to(DEVICE)
76
  processor = AutoProcessor.from_pretrained(MODEL_PATH, use_fast=True)
77
  print(f"Qwen model loaded on {DEVICE}")
78
  except Exception as e:
 
72
  MODEL_PATH,
73
  torch_dtype=torch.bfloat16 if DEVICE == "cuda" else torch.float32,
74
  device_map="auto" if DEVICE == "cuda" else None
75
+ )
76
  processor = AutoProcessor.from_pretrained(MODEL_PATH, use_fast=True)
77
  print(f"Qwen model loaded on {DEVICE}")
78
  except Exception as e: