AthuKawale commited on
Commit
925d43b
·
1 Parent(s): 66d1913

check update

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,5 +1,8 @@
1
  from peft import PeftModel, PeftConfig
2
 
 
 
 
3
  from datasets import load_dataset
4
  dataset_name = "timdettmers/openassistant-guanaco"
5
  dataset = load_dataset(dataset_name, split="train")
 
1
  from peft import PeftModel, PeftConfig
2
 
3
+ torch.cuda.is_available()
4
+ print("executed successfully")
5
+
6
  from datasets import load_dataset
7
  dataset_name = "timdettmers/openassistant-guanaco"
8
  dataset = load_dataset(dataset_name, split="train")