Spaces:

rhfeiyang
/

Art-Free-Diffusion

Running on Zero

rhfeiyang commited on 3 days ago

Commit

cf66859

•

1 Parent(s): 5d89dc0

update

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -354,7 +354,7 @@ def inference(network: LoRANetwork, tokenizer: CLIPTokenizer, text_encoder: CLIP
                 latent_model_input = noise_scheduler.scale_model_input(latent_model_input, timestep=t).to(weight_dtype)
                 # predict the noise residual
                 with network:
-                    print(f"dtype: {latent_model_input.dtype}, {text_embedding.dtype}, t={t}")
                     noise_pred = unet(latent_model_input, t , encoder_hidden_states=text_embedding).sample
                 # perform guidance

                 latent_model_input = noise_scheduler.scale_model_input(latent_model_input, timestep=t).to(weight_dtype)
                 # predict the noise residual
                 with network:
+                    # print(f"dtype: {latent_model_input.dtype}, {text_embedding.dtype}, t={t}")
                     noise_pred = unet(latent_model_input, t , encoder_hidden_states=text_embedding).sample
                 # perform guidance