Update app.py
Browse files
app.py
CHANGED
@@ -126,27 +126,27 @@ def plex(prompt,neg_prompt,modil,stips,scaly,nut,wei,hei,los,loca,gaul,progress=
|
|
126 |
except:
|
127 |
print("no card")
|
128 |
try:
|
129 |
-
pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.bfloat16, variant="fp16", use_safetensors=True, safety_checker=
|
130 |
except:
|
131 |
gc.collect()
|
132 |
try:
|
133 |
-
pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float32, variant="fp32", use_safetensors=True, safety_checker=
|
134 |
except:
|
135 |
gc.collect()
|
136 |
try:
|
137 |
-
pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.bfloat16, variant="fp16", use_safetensors=False, safety_checker=
|
138 |
except:
|
139 |
gc.collect()
|
140 |
try:
|
141 |
-
pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float32, variant="fp32", use_safetensors=False, safety_checker=
|
142 |
except:
|
143 |
gc.collect()
|
144 |
try:
|
145 |
-
pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float, variant=None, use_safetensors=True, safety_checker=
|
146 |
except:
|
147 |
gc.collect()
|
148 |
try:
|
149 |
-
pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float, variant=None, use_safetensors=False, safety_checker=
|
150 |
except:
|
151 |
print("no pipe")
|
152 |
if los:
|
|
|
126 |
except:
|
127 |
print("no card")
|
128 |
try:
|
129 |
+
pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.bfloat16, variant="fp16", use_safetensors=True, safety_checker=None))
|
130 |
except:
|
131 |
gc.collect()
|
132 |
try:
|
133 |
+
pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float32, variant="fp32", use_safetensors=True, safety_checker=None))
|
134 |
except:
|
135 |
gc.collect()
|
136 |
try:
|
137 |
+
pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.bfloat16, variant="fp16", use_safetensors=False, safety_checker=None))
|
138 |
except:
|
139 |
gc.collect()
|
140 |
try:
|
141 |
+
pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float32, variant="fp32", use_safetensors=False, safety_checker=None))
|
142 |
except:
|
143 |
gc.collect()
|
144 |
try:
|
145 |
+
pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float, variant=None, use_safetensors=True, safety_checker=None))
|
146 |
except:
|
147 |
gc.collect()
|
148 |
try:
|
149 |
+
pipe=accelerator.prepare(DiffusionPipeline.from_pretrained(""+modil+"",torch_dtype=torch.float, variant=None, use_safetensors=False, safety_checker=None))
|
150 |
except:
|
151 |
print("no pipe")
|
152 |
if los:
|