j-min commited on
Commit
0d9c9f0
β€’
1 Parent(s): 6fc6364

load directly to self attribute

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -62,7 +62,8 @@ class Instance:
62
 
63
  # # xformers
64
  # pipe.enable_xformers_memory_efficient_attention()
65
- return pipe
 
66
 
67
  instance = Instance()
68
  instance.load_model()
 
62
 
63
  # # xformers
64
  # pipe.enable_xformers_memory_efficient_attention()
65
+
66
+ self.loaded_model = pipe
67
 
68
  instance = Instance()
69
  instance.load_model()