pradeep6kumar2024 commited on
Commit
c6e01dd
·
1 Parent(s): 3cbb88b

added readme and app.py

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: Phi-2 QLoRA Assistant Demo (CPU-Optimized)
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(