aifeifei798 commited on
Commit
5ff2701
·
verified ·
1 Parent(s): a670616

Upload feifeiprompt.py

Browse files
Files changed (1) hide show
  1. feifeilib/feifeiprompt.py +1 -0
feifeilib/feifeiprompt.py CHANGED
@@ -2,6 +2,7 @@ import numpy as np
2
  import random
3
  from extras.expansion import FooocusExpansion
4
  import re
 
5
 
6
  expansion = FooocusExpansion()
7
  with open("artist.txt", "r") as file:
 
2
  import random
3
  from extras.expansion import FooocusExpansion
4
  import re
5
+ import torch
6
 
7
  expansion = FooocusExpansion()
8
  with open("artist.txt", "r") as file: