Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,6 @@ tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|
|
| 10 |
# Load the Lora model
|
| 11 |
finetuned_model = PeftModel.from_pretrained(finetuned_model, hf_repo)
|
| 12 |
|
| 13 |
-
|
| 14 |
@torch.no_grad()
|
| 15 |
def infer(text):
|
| 16 |
if torch.cuda.is_available():
|
|
|
|
| 10 |
# Load the Lora model
|
| 11 |
finetuned_model = PeftModel.from_pretrained(finetuned_model, hf_repo)
|
| 12 |
|
|
|
|
| 13 |
@torch.no_grad()
|
| 14 |
def infer(text):
|
| 15 |
if torch.cuda.is_available():
|