Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -113,8 +113,8 @@ def load_and_prepare_model():
|
|
113 |
#unetX = UNet2DConditionModel.from_pretrained('ford442/RealVisXL_V5.0_BF16',subfolder='unet').to(torch.bfloat16) # ,use_safetensors=True FAILS
|
114 |
#sched = EulerAncestralDiscreteScheduler.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='scheduler',beta_schedule="scaled_linear", steps_offset=1,timestep_spacing="trailing"))
|
115 |
#sched = EulerAncestralDiscreteScheduler.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='scheduler', steps_offset=1,timestep_spacing="trailing")
|
116 |
-
sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1,use_karras_sigmas=True)
|
117 |
-
|
118 |
#sched = DPMSolverSDEScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler')
|
119 |
#pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0").to(torch.bfloat16)
|
120 |
#pipeX = StableDiffusionXLPipeline.from_pretrained("ford442/Juggernaut-XI-v11-fp32",use_safetensors=True)
|
@@ -188,8 +188,9 @@ def load_and_prepare_model():
|
|
188 |
pipe.set_adapters(["skin"], adapter_weights=[0.5]) #pipe.unet.set_default_attn_processor()
|
189 |
|
190 |
pipe.vae.set_default_attn_processor()
|
191 |
-
pipe.to(device)
|
192 |
-
pipe.to(
|
|
|
193 |
|
194 |
print(f'Pipeline: ')
|
195 |
#print(f'_optional_components: {pipe._optional_components}')
|
@@ -205,7 +206,7 @@ def load_and_prepare_model():
|
|
205 |
return pipe
|
206 |
|
207 |
pipe = load_and_prepare_model()
|
208 |
-
MAX_SEED = np.iinfo(np.
|
209 |
|
210 |
neg_prompt_2 = " 'non-photorealistic':1.5, 'unrealistic skin','unattractive face':1.3, 'low quality':1.1, ('dull color scheme', 'dull colors', 'digital noise':1.2),'amateurish', 'poorly drawn face':1.3, 'poorly drawn', 'distorted face', 'low resolution', 'simplistic' "
|
211 |
|
|
|
113 |
#unetX = UNet2DConditionModel.from_pretrained('ford442/RealVisXL_V5.0_BF16',subfolder='unet').to(torch.bfloat16) # ,use_safetensors=True FAILS
|
114 |
#sched = EulerAncestralDiscreteScheduler.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='scheduler',beta_schedule="scaled_linear", steps_offset=1,timestep_spacing="trailing"))
|
115 |
#sched = EulerAncestralDiscreteScheduler.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='scheduler', steps_offset=1,timestep_spacing="trailing")
|
116 |
+
#sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1,use_karras_sigmas=True)
|
117 |
+
sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear")
|
118 |
#sched = DPMSolverSDEScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler')
|
119 |
#pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0").to(torch.bfloat16)
|
120 |
#pipeX = StableDiffusionXLPipeline.from_pretrained("ford442/Juggernaut-XI-v11-fp32",use_safetensors=True)
|
|
|
188 |
pipe.set_adapters(["skin"], adapter_weights=[0.5]) #pipe.unet.set_default_attn_processor()
|
189 |
|
190 |
pipe.vae.set_default_attn_processor()
|
191 |
+
pipe.to(device, torch.bfloat16)
|
192 |
+
#pipe.to(device)
|
193 |
+
#pipe.to(torch.bfloat16)
|
194 |
|
195 |
print(f'Pipeline: ')
|
196 |
#print(f'_optional_components: {pipe._optional_components}')
|
|
|
206 |
return pipe
|
207 |
|
208 |
pipe = load_and_prepare_model()
|
209 |
+
MAX_SEED = np.iinfo(np.int32).max
|
210 |
|
211 |
neg_prompt_2 = " 'non-photorealistic':1.5, 'unrealistic skin','unattractive face':1.3, 'low quality':1.1, ('dull color scheme', 'dull colors', 'digital noise':1.2),'amateurish', 'poorly drawn face':1.3, 'poorly drawn', 'distorted face', 'low resolution', 'simplistic' "
|
212 |
|