Hugging Face's logo

Spaces:
plotari2025
/
interior-ai-designer
Sleeping

App Files Community
broyang commited on
Commit
f17fd1b
·
verified ·
1 Parent(s): 809d317

cpu offload

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -84,7 +84,8 @@ pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_unc
84
  pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_NakedHoodie.pt", token="HDA_NakedHoodie")
85
  pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_NunDress.pt", token="HDA_NunDress")
86
  pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Shibari.pt", token="HDA_Shibari")
87
- pipe.to("cuda")
 
88
 
89
  print("loading preprocessor")
90
  from preprocess import Preprocessor
 
84
  pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_NakedHoodie.pt", token="HDA_NakedHoodie")
85
  pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_NunDress.pt", token="HDA_NunDress")
86
  pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Shibari.pt", token="HDA_Shibari")
87
+ # pipe.to("cuda")
88
+ pipe.enable_model_cpu_offload()
89
 
90
  print("loading preprocessor")
91
  from preprocess import Preprocessor