Alkhalaf commited on
Commit
a90fba5
·
verified ·
1 Parent(s): fc79e83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ from huggingface_hub import login
5
  import torch
6
  import os
7
  hf_token = os.getenv("llama")
 
8
  #login(hf_token)
9
  # Model and adapter paths
10
  model_name = "unsloth/llama-3.2-1b-instruct-bnb-4bit" # Base model
 
5
  import torch
6
  import os
7
  hf_token = os.getenv("llama")
8
+ assert hf_token is not None, "Token is missing! Make sure 'llama' is set in the environment."
9
  #login(hf_token)
10
  # Model and adapter paths
11
  model_name = "unsloth/llama-3.2-1b-instruct-bnb-4bit" # Base model