ashawkey commited on
Commit
0cb8c0e
1 Parent(s): f6e1b58

fix huggingface token, add requirements

Browse files
Files changed (2) hide show
  1. nerf/sd.py +1 -1
  2. requirements.txt +1 -0
nerf/sd.py CHANGED
@@ -16,7 +16,7 @@ class StableDiffusion(nn.Module):
16
 
17
  try:
18
  with open('./TOKEN', 'r') as f:
19
- self.token = f.read()
20
  print(f'[INFO] successfully loaded hugging face user token!')
21
  except FileNotFoundError as e:
22
  print(e)
 
16
 
17
  try:
18
  with open('./TOKEN', 'r') as f:
19
+ self.token = f.read().replace('\n', '') # remove the last \n!
20
  print(f'[INFO] successfully loaded hugging face user token!')
21
  except FileNotFoundError as e:
22
  print(e)
requirements.txt CHANGED
@@ -14,4 +14,5 @@ pysdf
14
  dearpygui
15
  scipy
16
  diffusers
 
17
  xatlas
 
14
  dearpygui
15
  scipy
16
  diffusers
17
+ transformers
18
  xatlas