[email protected] commited on
Commit
f432656
1 Parent(s): 07f142a

put other model in comment

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -31,8 +31,13 @@ if os.environ.get("HF_TOKEN") is None:
31
  )
32
 
33
  # sets the main paremeters
 
 
 
 
34
  hugging_face_model_id = "eltorio/Llama-3.2-3B-appreciation"
35
  base_model_path = "meta-llama/Llama-3.2-3B-Instruct"
 
36
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
37
 
38
  # Define the title, description, and device description for the Gradio interface
 
31
  )
32
 
33
  # sets the main paremeters
34
+
35
+ # hugging_face_model_id = "eltorio/Llama-3.1-8B-appreciation"
36
+ # base_model_path = "meta-llama/Llama-3.1-8B-Instruct"
37
+
38
  hugging_face_model_id = "eltorio/Llama-3.2-3B-appreciation"
39
  base_model_path = "meta-llama/Llama-3.2-3B-Instruct"
40
+
41
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
42
 
43
  # Define the title, description, and device description for the Gradio interface