JoPmt commited on
Commit
faeb036
1 Parent(s): c030f70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -123,22 +123,27 @@ def plex(prompt,neg_prompt,modil,stips,scaly,nut,wei,hei,los,loca,gaul,progress=
123
  try:
124
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.bfloat16, variant="fp16", use_safetensors=True, safety_checker=False))
125
  except:
 
126
  try:
127
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float32, variant="fp32", use_safetensors=True, safety_checker=False))
128
  except:
 
129
  try:
130
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.bfloat16, variant="fp16", use_safetensors=False, safety_checker=False))
131
  except:
 
132
  try:
133
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float32, variant="fp32", use_safetensors=False, safety_checker=False))
134
  except:
 
135
  try:
136
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float, variant=None, use_safetensors=True, safety_checker=False))
137
  except:
 
138
  try:
139
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float, variant=None, use_safetensors=False, safety_checker=False))
140
- except:
141
- print("no pipe")
142
  if los:
143
  try:
144
  lrda=ModelCard.load(""+los+"")
 
123
  try:
124
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.bfloat16, variant="fp16", use_safetensors=True, safety_checker=False))
125
  except:
126
+ gc.collect()
127
  try:
128
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float32, variant="fp32", use_safetensors=True, safety_checker=False))
129
  except:
130
+ gc.collect()
131
  try:
132
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.bfloat16, variant="fp16", use_safetensors=False, safety_checker=False))
133
  except:
134
+ gc.collect()
135
  try:
136
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float32, variant="fp32", use_safetensors=False, safety_checker=False))
137
  except:
138
+ gc.collect()
139
  try:
140
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float, variant=None, use_safetensors=True, safety_checker=False))
141
  except:
142
+ gc.collect()
143
  try:
144
  pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float, variant=None, use_safetensors=False, safety_checker=False))
145
+ except:
146
+ print("no pipe")
147
  if los:
148
  try:
149
  lrda=ModelCard.load(""+los+"")