Spaces:

ruslanmv
/

Medical-Llama3-Chatbot

Running on Zero

ruslanmv commited on May 15

Commit

4771e5d

•

1 Parent(s): f4fd3a7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,6 +3,7 @@ import os
 os.system("git clone https://github.com/TimDettmers/bitsandbytes.git")
 os.system("cd bitsandbytes/ && pip install -r requirements-dev.txt && cmake -DCOMPUTE_BACKEND=cuda -S . && make && pip install .")
 # Check if GPU is available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(f"Using device: {device}")

 os.system("git clone https://github.com/TimDettmers/bitsandbytes.git")
 os.system("cd bitsandbytes/ && pip install -r requirements-dev.txt && cmake -DCOMPUTE_BACKEND=cuda -S . && make && pip install .")
 # Check if GPU is available
+import torch
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(f"Using device: {device}")