reshinthadith commited on
Commit
d9b7ce5
1 Parent(s): f541eb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import time
5
  import numpy as np
6
  from torch.nn import functional as F
7
  import os
8
- token_key = os.environ.get(HUGGING_FACE_HUB_TOKEN)
9
 
10
  if torch.cuda.is_available():
11
  m = AutoModelForCausalLM.from_pretrained("stabilityai/stablelm-tuned-alpha-7b",use_auth_token=token_key, torch_dtype=torch.float16).cuda()
 
5
  import numpy as np
6
  from torch.nn import functional as F
7
  import os
8
+ token_key = True#os.environ.get("HUGGING_FACE_HUB_TOKEN")
9
 
10
  if torch.cuda.is_available():
11
  m = AutoModelForCausalLM.from_pretrained("stabilityai/stablelm-tuned-alpha-7b",use_auth_token=token_key, torch_dtype=torch.float16).cuda()