nttwt1597 commited on
Commit
57c5e2b
·
verified ·
1 Parent(s): 2836cb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -20,10 +20,10 @@ import gradio as gr
20
  model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
21
  peft_model_adapter_id = "nttwt1597/test_v2"
22
 
23
- model_directory = "./model/"
24
 
25
- device = "cuda" if torch.cuda.is_available() else "cpu"
26
- print("Using:", device)
27
 
28
  #Commented out IPython magic to ensure Python compatibility.
29
  #%%capture
 
20
  model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
21
  peft_model_adapter_id = "nttwt1597/test_v2"
22
 
23
+ # model_directory = "./model/"
24
 
25
+ # device = "cuda" if torch.cuda.is_available() else "cpu"
26
+ # print("Using:", device)
27
 
28
  #Commented out IPython magic to ensure Python compatibility.
29
  #%%capture