minhdang commited on
Commit
5cf089b
1 Parent(s): 47a6c71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,8 +4,9 @@ import transformers
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig,AwqConfig
5
  import torch
6
  import os
7
- # from huggingface_hub import login
8
- # login(key)
 
9
 
10
 
11
  nf4_config = BitsAndBytesConfig(
 
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig,AwqConfig
5
  import torch
6
  import os
7
+ key = os.environ.get("key")
8
+ from huggingface_hub import login
9
+ login(key)
10
 
11
 
12
  nf4_config = BitsAndBytesConfig(