Sahibsingh12 commited on
Commit
1345934
·
1 Parent(s): 0445f94

files changed

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ install('transformers')
10
  install('torch')
11
 
12
  from transformers import AutoModelForCausalLM, AutoTokenizer
13
-
14
  model = AutoModelForCausalLM.from_pretrained("Sahibsingh12/phi-1-5-finetuned-cazton_complete", trust_remote_code=True, torch_dtype=torch.float32)
15
  tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1_5", trust_remote_code=True)
16
 
 
10
  install('torch')
11
 
12
  from transformers import AutoModelForCausalLM, AutoTokenizer
13
+ import torch
14
  model = AutoModelForCausalLM.from_pretrained("Sahibsingh12/phi-1-5-finetuned-cazton_complete", trust_remote_code=True, torch_dtype=torch.float32)
15
  tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1_5", trust_remote_code=True)
16