Yurii Paniv
commited on
Commit
•
dc34b69
1
Parent(s):
8ace6ee
Fix peft
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 28 |
|
| 29 |
model = PeftModel.from_pretrained(model, lora_name, torch_device="cpu")
|
| 30 |
|
| 31 |
-
|
| 32 |
|
| 33 |
from transformers import StoppingCriteriaList, StopStringCriteria, TextIteratorStreamer
|
| 34 |
from threading import Thread
|
|
|
|
| 28 |
|
| 29 |
model = PeftModel.from_pretrained(model, lora_name, torch_device="cpu")
|
| 30 |
|
| 31 |
+
model = model.to("cuda")
|
| 32 |
|
| 33 |
from transformers import StoppingCriteriaList, StopStringCriteria, TextIteratorStreamer
|
| 34 |
from threading import Thread
|