Spaces:
Sleeping
Sleeping
Henry Hommel commited on
Commit ·
fdd45c2
1
Parent(s): 2c362df
Update clbaseimplementation.py
Browse files- clbaseimplementation.py +0 -3
clbaseimplementation.py
CHANGED
|
@@ -9,9 +9,6 @@ import chainlit as cl
|
|
| 9 |
import torch
|
| 10 |
import time
|
| 11 |
import tempfile
|
| 12 |
-
from numba import cuda
|
| 13 |
-
device = cuda.get_current_device()
|
| 14 |
-
device.reset()
|
| 15 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
|
| 16 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
| 17 |
from langchain.vectorstores import Chroma
|
|
|
|
| 9 |
import torch
|
| 10 |
import time
|
| 11 |
import tempfile
|
|
|
|
|
|
|
|
|
|
| 12 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
|
| 13 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
| 14 |
from langchain.vectorstores import Chroma
|