Spaces:
Sleeping
Sleeping
Commit
•
a69cdec
1
Parent(s):
a2d98e7
updated app.py
Browse files
app.py
CHANGED
|
@@ -5,8 +5,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
| 5 |
|
| 6 |
|
| 7 |
#Loading model
|
| 8 |
-
quantization_config = BitsAndBytesConfig(load_in_4bit=True)
|
| 9 |
-
|
| 10 |
model_path = "parsanna17/finetune_starcoder2_with_R_data"
|
| 11 |
checkpoint = "bigcode/starcoder2-3b"
|
| 12 |
config = PeftConfig.from_pretrained(model_path)
|
|
|
|
| 5 |
|
| 6 |
|
| 7 |
#Loading model
|
|
|
|
|
|
|
| 8 |
model_path = "parsanna17/finetune_starcoder2_with_R_data"
|
| 9 |
checkpoint = "bigcode/starcoder2-3b"
|
| 10 |
config = PeftConfig.from_pretrained(model_path)
|