tight-inversion commited on
Commit
c67f10c
·
1 Parent(s): 0f1b614
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ class FluxGenerator:
34
  self.device = torch.device('cuda')
35
  self.offload = False
36
  self.model_name = 'flux-dev'
37
- self.model, self.ae, self.t5, self.clip, self.nsfw_classifier = get_models(
38
  self.model_name,
39
  device=self.device,
40
  offload=self.offload,
 
34
  self.device = torch.device('cuda')
35
  self.offload = False
36
  self.model_name = 'flux-dev'
37
+ self.model, self.ae, self.t5, self.clip = get_models(
38
  self.model_name,
39
  device=self.device,
40
  offload=self.offload,