Hugging Face's logo

Spaces:
relik-ie
/
testing
Sleeping

App Files Community
relik-ie commited on
Commit
ab2532f
•
1 Parent(s): 035b09f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ For more information on `huggingface_hub` Inference API support, please check th
13
  # tokenizer = AutoTokenizer.from_pretrained("sapienzanlp/Minerva-7B-base-v1.0-step500000-instruct-smol")
14
  # model = AutoModelForCausalLM.from_pretrained("sapienzanlp/Minerva-7B-base-v1.0-step500000-instruct-smol", torch_dtype=torch.float16)
15
  tokenizer = AutoTokenizer.from_pretrained("sapienzanlp/Minerva-7B-base-v1.0-step500000-instruct-fix_digit", revision="step500")
16
- model = AutoModelForCausalLM.from_pretrained("sapienzanlp/Minerva-7B-base-v1.0-step500000-instruct-fix_digit", revision="step500" torch_dtype=torch.float16)
17
 
18
  model = model.to('cuda:0')
19
 
 
13
  # tokenizer = AutoTokenizer.from_pretrained("sapienzanlp/Minerva-7B-base-v1.0-step500000-instruct-smol")
14
  # model = AutoModelForCausalLM.from_pretrained("sapienzanlp/Minerva-7B-base-v1.0-step500000-instruct-smol", torch_dtype=torch.float16)
15
  tokenizer = AutoTokenizer.from_pretrained("sapienzanlp/Minerva-7B-base-v1.0-step500000-instruct-fix_digit", revision="step500")
16
+ model = AutoModelForCausalLM.from_pretrained("sapienzanlp/Minerva-7B-base-v1.0-step500000-instruct-fix_digit", revision="step500", torch_dtype=torch.float16)
17
 
18
  model = model.to('cuda:0')
19