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(
|