Tonic commited on
Commit
88023ca
·
1 Parent(s): b2dd6fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
14
  model_name = "microsoft/Orca-2-13b"
15
  # offload_folder = './model_weights'
16
 
17
- if not os.path.exists(offload_folder):
18
- os.makedirs(offload_folder)
19
 
20
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
21
  model = AutoModelForCausalLM.from_pretrained(model_name)
 
14
  model_name = "microsoft/Orca-2-13b"
15
  # offload_folder = './model_weights'
16
 
17
+ # if not os.path.exists(offload_folder):
18
+ # os.makedirs(offload_folder)
19
 
20
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
21
  model = AutoModelForCausalLM.from_pretrained(model_name)