Omnibus commited on
Commit
bedc7d9
·
verified ·
1 Parent(s): 14c7a51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def list_files(model_name):
14
 
15
  def load_bin(model_name,file_name):
16
  r = requests.get(f'https://huggingface.co/{model_name}/resolve/main/{file_name}')
17
- result=torch.frombuffer(r,dtype=get_default_dtype())
18
 
19
  return result
20
  #with open()
 
14
 
15
  def load_bin(model_name,file_name):
16
  r = requests.get(f'https://huggingface.co/{model_name}/resolve/main/{file_name}')
17
+ result=torch.frombuffer(r,dtype=torch.get_default_dtype())
18
 
19
  return result
20
  #with open()