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
|