vansh-myth commited on
Commit
a37efe3
·
verified ·
1 Parent(s): 6fb14d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -19,6 +19,8 @@ model = PeftModel.from_pretrained(
19
  device_map = "auto"
20
  )
21
 
 
 
22
  # Optional: move model to correct device
23
  if torch.cuda.is_available():
24
  model = model.to("cuda")
 
19
  device_map = "auto"
20
  )
21
 
22
+ FastLanguageModel.for_inference(model)
23
+
24
  # Optional: move model to correct device
25
  if torch.cuda.is_available():
26
  model = model.to("cuda")