ChatBot / app.py
AthuKawale's picture
changed for error libcuda.so
18d159f
raw
history blame
325 Bytes
from peft import PeftModel, PeftConfig
import torch
from datasets import load_dataset
torch.cuda.is_available()
print("executed successfully")
dataset_name = "timdettmers/openassistant-guanaco"
dataset = load_dataset(dataset_name, split="train")
echo $LD_LIBRARY_PATH #path
sudo find /usr/ -name 'libcuda.so.*' #version