Spaces:
Sleeping
Sleeping
Commit
·
c6e01dd
1
Parent(s):
3cbb88b
added readme and app.py
Browse files
README.md
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
---
|
| 2 |
-
title: Phi-2 QLoRA Assistant
|
| 3 |
emoji: 🤖
|
| 4 |
colorFrom: blue
|
| 5 |
colorTo: purple
|
|
@@ -77,6 +77,7 @@ This demo is released under the MIT License.
|
|
| 77 |
```python
|
| 78 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 79 |
from peft import PeftModel
|
|
|
|
| 80 |
|
| 81 |
# Load base model and adapter (CPU optimized)
|
| 82 |
base_model = AutoModelForCausalLM.from_pretrained(
|
|
|
|
| 1 |
---
|
| 2 |
+
title: Phi-2 QLoRA Assistant
|
| 3 |
emoji: 🤖
|
| 4 |
colorFrom: blue
|
| 5 |
colorTo: purple
|
|
|
|
| 77 |
```python
|
| 78 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 79 |
from peft import PeftModel
|
| 80 |
+
import torch
|
| 81 |
|
| 82 |
# Load base model and adapter (CPU optimized)
|
| 83 |
base_model = AutoModelForCausalLM.from_pretrained(
|