Omnibus commited on
Commit
0563105
·
verified ·
1 Parent(s): 589645a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,10 +15,10 @@ def list_files(model_name):
15
 
16
  def load_bin(model_name,file_name):
17
  r = requests.get(f'https://huggingface.co/{model_name}/resolve/main/{file_name}')
18
- print(r.content)
19
  print("#######")
20
  print(dir(r))
21
- result=torch.frombuffer(r.content,dtype=torch.get_default_dtype())
22
  #result = torch.frombuffer(io.BytesIO(),dtype=torch.get_default_dtype())
23
  #result = torch.from_numpy(r.content)
24
  return result
 
15
 
16
  def load_bin(model_name,file_name):
17
  r = requests.get(f'https://huggingface.co/{model_name}/resolve/main/{file_name}')
18
+ #print(r.content)
19
  print("#######")
20
  print(dir(r))
21
+ result=torch.frombuffer(r.content,dtype=torch.get_default_dtype(),offset=3)
22
  #result = torch.frombuffer(io.BytesIO(),dtype=torch.get_default_dtype())
23
  #result = torch.from_numpy(r.content)
24
  return result