Spaces:
Sleeping
Sleeping
Commit ·
201538e
1
Parent(s): 9b1b576
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,9 +6,6 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
| 6 |
from functools import partial
|
| 7 |
import gc
|
| 8 |
|
| 9 |
-
from datasets import load_dataset
|
| 10 |
-
dataset = load_dataset(path="wikitext", name="wikitext-103-v1", split="train")
|
| 11 |
-
|
| 12 |
|
| 13 |
|
| 14 |
# core quantization method (simulated quantization)
|
|
|
|
| 6 |
from functools import partial
|
| 7 |
import gc
|
| 8 |
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
|
| 11 |
# core quantization method (simulated quantization)
|