Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -18,6 +18,12 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
| 18 |
from peft import PeftModel
|
| 19 |
|
| 20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
MODEL_ID = "Qwen/Qwen2-1.5B-Instruct"
|
| 22 |
ADAPTER_DIR = os.environ.get("ADAPTER_DIR", "models/qlora-adapter")
|
| 23 |
|
|
|
|
| 18 |
from peft import PeftModel
|
| 19 |
|
| 20 |
|
| 21 |
+
|
| 22 |
+
print("Torch:", torch.__version__)
|
| 23 |
+
print("Transformers:", transformers.__version__)
|
| 24 |
+
print("PEFT:", peft.__version__)
|
| 25 |
+
|
| 26 |
+
|
| 27 |
MODEL_ID = "Qwen/Qwen2-1.5B-Instruct"
|
| 28 |
ADAPTER_DIR = os.environ.get("ADAPTER_DIR", "models/qlora-adapter")
|
| 29 |
|