Spaces:
Build error
Build error
Update app.py
Browse files
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")
|