Omnibus commited on
Commit
0659d8a
·
verified ·
1 Parent(s): b29cd11

Update app.py

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