mouryachinta commited on
Commit
f927f46
β€’
1 Parent(s): 1cb4ed0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -6,6 +6,8 @@ from transformers import (
6
  TrainingArguments,
7
  pipeline
8
  )
 
 
9
  from peft import LoraConfig
10
 
11
  base_model_name = "mouryachinta/llama-2-7b-mourya"
 
6
  TrainingArguments,
7
  pipeline
8
  )
9
+ import torch
10
+ from langchain.llms.base import LLM
11
  from peft import LoraConfig
12
 
13
  base_model_name = "mouryachinta/llama-2-7b-mourya"