fix huggingface token, add requirements
Browse files- nerf/sd.py +1 -1
- 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
|